Home
last modified time | relevance | path

Searched defs:mul_node (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dstrip_unused_nodes_test.cc138 NodeDef* mul_node = graph_def.add_node(); in TestSimplePlaceholder() local
181 NodeDef* mul_node = graph_def.add_node(); in TestPlaceholderDefaultArgs() local
227 NodeDef* mul_node = graph_def.add_node(); in TestPlaceholderNamedArgs() local
Dfold_batch_norms.cc52 const NodeDef& mul_node = match.node; in FoldBatchNorms() local
Dtransform_utils_test.cc705 NodeDef* mul_node = graph_def.mutable_node()->Add(); in TestFindInvalidInputs() local
753 NodeDef* mul_node = invalid_graph_def.mutable_node()->Add(); in TestIsGraphValid() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dfuse_mul_to_conv_test.cc64 auto mul_node = graph.NewNode(); in TEST() local
115 auto mul_node = graph.NewNode(); in TEST() local
Dfuse_mul_to_conv.cc49 auto& mul_node = *sequence[1]; in ApplyToNodesSequence() local
113 auto& mul_node = *sequence[0]; in ApplyToNodesSequence() local
/external/tensorflow/tensorflow/core/grappler/
Dutils_test.cc357 const NodeDef* mul_node = node_map.GetNode("mul"); in TEST_F() local
394 const NodeDef* mul_node = node_map.GetNode("mul"); in TEST_F() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc464 auto mul_node = node_map_output.GetNode("mul"); in TEST_F() local
843 auto mul_node = node_map.GetNode("mul"); in TEST_F() local
860 auto mul_node = node_map.GetNode("mul"); in TEST_F() local
880 auto mul_node = node_map.GetNode("mul"); in TEST_F() local
900 auto mul_node = node_map.GetNode("mul"); in TEST_F() local
917 auto mul_node = node_map.GetNode("mul"); in TEST_F() local
941 auto mul_node = node_map.GetNode("mul"); in TEST_F() local
1234 auto mul_node = node_map.GetNode("mul"); in TEST_F() local
Darithmetic_optimizer_test.cc401 const NodeDef* mul_node = node_map.GetNode(HoistMulName("Add_6")); in TEST_F() local