| D | array_grad_test.cc | 35 void RunTest(const Output& x, const TensorShape& x_shape, const Output& y, in RunTest() function in tensorflow::__anon62bed55d0111::ArrayGradTest 44 void RunTest(const OutputList& xs, const std::vector<TensorShape>& x_shapes, in RunTest() function in tensorflow::__anon62bed55d0111::ArrayGradTest 63 RunTest(xs, {x_shape, x_shape}, {y}, {y_shape}); in TEST_F() 73 RunTest(xs, {x_shape, x_shape}, {y}, {y_shape}); in TEST_F() 82 RunTest({x}, {x_shape}, y.output, y_shapes); in TEST_F() 91 RunTest({x}, {x_shape}, y.output, y_shapes); in TEST_F() 98 RunTest(x, shape, y, shape); in TEST_F() 108 RunTest({x}, {x_shape}, y.output, {y_shape, y_shape}); in TEST_F() 115 RunTest({x}, {x_shape}, y.output, in TEST_F() 124 RunTest(x, x_shape, y, y_shape); in TEST_F() [all …]
|