| D | layout_optimizer_test.cc | 61 Output SimpleConv2D(tensorflow::Scope* s, int input_size, int filter_size, in SimpleConv2D() function in tensorflow::grappler::__anon47ae83710111::LayoutOptimizerTest
63 return SimpleConv2D<T>(s, input_size, filter_size, padding, ""); in SimpleConv2D()
67 Output SimpleConv2D(tensorflow::Scope* s, int input_size, int filter_size, in SimpleConv2D() function in tensorflow::grappler::__anon47ae83710111::LayoutOptimizerTest
289 auto conv = SimpleConv2D(&s, 2, 1, "SAME"); in TEST_F()
302 auto conv = SimpleConv2D(&s, 2, 1, "SAME"); in TEST_F()
315 auto conv = SimpleConv2D(&s, 2, 2, "VALID"); in TEST_F()
328 auto conv = SimpleConv2D(&s, 2, 2, "SAME"); in TEST_F()
341 auto conv = SimpleConv2D(&s, 4, 2, "VALID"); in TEST_F()
354 auto conv = SimpleConv2D(&s, 4, 2, "EXPLICIT"); in TEST_F()
367 auto conv = SimpleConv2D<int32>(&s, 4, 2, "EXPLICIT"); in TEST_F()
[all …]
|