| D | atest.cpp | 14 y_tensor = tensor({-10, 1, 0, -1, 10}); in SetUp()
22 Tensor x_tensor, y_tensor; member in atest
49 const Tensor& y_tensor, in unit_binary_ops_test() argument
54 func(out_tensor, x_tensor.to(dtype), y_tensor.to(dtype), args...); in unit_binary_ops_test()
79 const Tensor& y_tensor, in run_binary_ops_test() argument
85 unit_binary_ops_test(func, x_tensor, y_tensor, exp, kInt, args...); in run_binary_ops_test()
90 unit_binary_ops_test(func, x_tensor, y_tensor, exp, kFloat, args...); in run_binary_ops_test()
95 unit_binary_ops_test(func, x_tensor, y_tensor, exp, kBool, args...); in run_binary_ops_test()
190 add_out, x_tensor, y_tensor, exp_tensor, INTBOOL, 2); in TEST_F()
197 max_out, x_tensor, y_tensor, exp_tensor, INTBOOLFLOAT); in TEST_F()
[all …]
|