func @main( | |
%input : tensor<1x8x8x3xf32> {iree.identifier = "input"} | |
) -> ( | |
tensor<1x8x8x3xf32> {iree.identifier = "output"} | |
) attributes { iree.module.export } { | |
%result = mhlo.add %input, %input : tensor<1x8x8x3xf32> | |
return %result : tensor<1x8x8x3xf32> | |
} |