Searched refs:get_conv2d_nodedef (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 3111 auto get_conv2d_nodedef = in TEST_F() local
3140 NodeDef node_def = get_conv2d_nodedef(); in TEST_F()
3150 NodeDef node_def = get_conv2d_nodedef(); in TEST_F()
3160 NodeDef node_def = get_conv2d_nodedef(); in TEST_F()
3171 get_conv2d_nodedef({1, 1, 1, 1}, "SAME", "NCHW", {1, 1, 1}); in TEST_F()
3182 get_conv2d_nodedef({1, 1, 1, 1}, "SAME", "NCHW", {1, 2, 1, 1}); in TEST_F()
3193 get_conv2d_nodedef({1, 1, 1, 1}, "SAME", "NHWC", {1, 1, 1, 2}); in TEST_F()
3204 get_conv2d_nodedef({1, 1, 1, 1}, "SAME", "NHWC", {1, 1, 2, 1}, true); in TEST_F()
3217 get_conv2d_nodedef({1, 1, 1}, "SAME", "NCHW", {1, 1, 1, 1}); in TEST_F()
3228 get_conv2d_nodedef({1, 2, 1, 1}, "SAME", "NCHW", {1, 1, 1, 1}); in TEST_F()
[all …]
|