Searched refs:MakeConstNodeDef (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 128 NodeDef MakeConstNodeDef(const string& name, const std::vector<T>& vals, in MakeConstNodeDef() function
137 NodeDef MakeConstNodeDef(const string& name, const std::vector<T>& vals) { in MakeConstNodeDef() function
141 return MakeConstNodeDef(name, vals, shape); in MakeConstNodeDef()
424 NodeDef node_def = MakeConstNodeDef<float>("my_const", {1.0f, 2.0f}); in TEST_F()
1233 validator_inputs_[name] = MakeConstNodeDef<T>(name, values, shape); in AddTestWeights()
1417 NodeDef node_def = MakeConstNodeDef<double>("my_const", {}); in TEST_F()
|