Searched refs:MakeNodeDef (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 112 NodeDef MakeNodeDef(const string& name, const string& op, in MakeNodeDef() function
491 NodeDef node_def = MakeNodeDef("my_op", "MyOp", {}); in TEST_F()
512 node_def = MakeNodeDef("my_op", "FakeQuantWithMinMaxArgs", {}); in TEST_F()
590 NodeDef node_def = MakeNodeDef("my_op", "MyOp", {"my_input"}); in TEST_F()
670 NodeDef node_def = MakeNodeDef("my_op", "MyOp", {"my_input"}); in TEST_F()
1409 NodeDef node_def = MakeNodeDef("my_const", "Const", {"input"}); in TEST_F()
1430 NodeDef node_def = MakeNodeDef("my_transpose", "Transpose", {}); in TEST_F()
1492 NodeDef node_def = MakeNodeDef("my_reshape", "Reshape", {}); in TEST_F()
1579 NodeDef node_def = MakeNodeDef("my_matmul", "MatMul", {}); in TEST_F()
1723 NodeDef node_def = MakeNodeDef("my_biasadd", "BiasAdd", {}); in TEST_F()
[all …]
|