Searched refs:GetBinaryOpNodeDef (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1736 NodeDef GetBinaryOpNodeDef(const string& input_name_l, in GetBinaryOpNodeDef() function
1767 node_def = GetBinaryOpNodeDef<OpType>("weights", "input", dtype); in TestBinaryTensorOpWeightNoBroadcast()
1769 node_def = GetBinaryOpNodeDef<OpType>("input", "weights", dtype); in TestBinaryTensorOpWeightNoBroadcast()
1822 GetBinaryOpNodeDef<ops::Add>("input", "weights", dtype); in TestBinaryTensorOpWeightWithChannelWiseBroadcast()
1863 GetBinaryOpNodeDef<ops::Add>("input", "weights", dtype); in TestBinaryTensorOpWeightWithUniformlyBroadcast()
1902 GetBinaryOpNodeDef<OpType>("input", "weights", dtype); in TestBinaryTensorOpWeightFallback()
1955 GetBinaryOpNodeDef<OpType>("input1", "input2", dtype); in TestBinaryTensorOpTensor()
|