Searched refs:TestOpConverter (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1927 void TestOpConverter(const string& name, const NodeDef node_def, in TestOpConverter() function in tensorflow::tensorrt::convert::ParameterizedOpConverterTestBase
2255 TestOpConverter("my_batchnorm", node_def, node_input[0].dims, in TEST_P()
2316 TestOpConverter("my_transpose", node_def, p.expected_output_dims, p.status, in TEST_P()
2463 TestOpConverter("my_shape", node_def, p.expected_output_dims, p.status, in TEST_P()
2766 TestOpConverter("my_biasadd", node_def, dims_array, Status::OK(), in TEST_P()
2843 TestOpConverter("my_binary", node_def, {2, 2, 2}, Status::OK(), in TEST_P()
2894 test->TestOpConverter("my_addn", node_def, p.dimensions, in TestAddN()
3125 TestOpConverter("my_square", node_def, {1, 1, 20}, Status::OK(), Status::OK(), in TEST_P()
3398 TestOpConverter("my_unary", node_def, p.expected_output_dims, Status::OK(), in TEST_P()
3487 TestOpConverter("my_expanddims", node_def, p.expected_output_dims, p.status, in TEST_P()
[all …]
|