func.func @tensor_float() { | |
%0 = util.unfoldable_constant dense<[[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]> : tensor<2x3xf32> | |
%1 = util.unfoldable_constant dense<[[7.0, 8.0, 9.0]]> : tensor<1x3xf32> | |
%2 = util.unfoldable_constant dense<[1.0]> : tensor<1xf32> | |
%result = tosa.fully_connected %0, %1, %2 : (tensor<2x3xf32>, tensor<1x3xf32>, tensor<1xf32>) -> tensor<2x1xf32> | |
check.expect_eq_const(%result, dense<[[51.0], [123.0]]> : tensor<2x1xf32>) : tensor<2x1xf32> | |
return | |
} |