Searched refs:MakeNode2Arg (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/graph_analyzer/ |
| D | test_tools.h | 36 NodeDef MakeNode2Arg(const string& name, const string& opcode,
45 return MakeNode2Arg(name, "Mul", arg1, arg2); in MakeNodeMul()
51 return MakeNode2Arg(name, "AddN", arg1, arg2); in MakeNodeAddN()
56 return MakeNode2Arg(name, "Sub", arg1, arg2); in MakeNodeSub()
63 return MakeNode2Arg(name, "BroadcastGradientArgs", arg1, arg2); in MakeNodeBroadcastGradientArgs()
|
| D | test_tools.cc | 35 NodeDef MakeNode2Arg(const string& name, const string& opcode, in MakeNode2Arg() function
62 return MakeNode2Arg(name, "ShapeN", arg1, arg2); in MakeNodeShapeN()
69 return MakeNode2Arg(name, "IdentityN", arg1, arg2); in MakeNodeIdentityN()
|