Searched refs:MakeNodeDef (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fingerprint_op_test.cc | 34 Status MakeNodeDef(DataType dtype, NodeDef* node_def) { in MakeNodeDef() function
48 TF_RETURN_IF_ERROR(MakeNodeDef(data->dtype(), node_def())); in MakeFingerprintOp()
215 TF_ASSERT_OK(MakeNodeDef(DT_UINT8, &op.node_def)); in TEST()
226 TF_ASSERT_OK(MakeNodeDef(DT_FLOAT, &op.node_def)); in TEST()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 152 NodeDef MakeNodeDef(const string& name, const string& op, in MakeNodeDef() function
804 NodeDef node_def = MakeNodeDef("my_op", "MyOp", {"my_input"}); in TEST_F()
887 NodeDef node_def = MakeNodeDef("my_op", "MyOp", {"my_input"}); in TEST_F()
914 NodeDef dummy_node_def = MakeNodeDef("dummy_op", "DummyOp", {}); in TEST_F()
952 NodeDef dummy_node_def = MakeNodeDef("dummy_op", "DummyOp", {}); in TestPrepareTensorForShape()
2788 MakeNodeDef("my_add", "Add", {"weights1", "weights2"}, {{"T", dtype}}); in TEST_P()
2983 MakeNodeDef("my_quantize", "FakeQuantWithMinMaxArgs", {"input"}); in TEST_F()
|