Home
last modified time | relevance | path

Searched defs:mul_node (Results 1 – 13 of 13) 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/lite/delegates/gpu/common/tasks/special/
Dconv_pointwise.cc158 NodeContext mean_node, mul_node, slice_node; in GetOffset() local
221 NodeContext mul_node; in TryFusedPointwiseConv() local
/external/tensorflow/tensorflow/core/grappler/
Dutils_test.cc360 const NodeDef* mul_node = node_map.GetNode("mul"); in TEST_F() local
397 const NodeDef* mul_node = node_map.GetNode("mul"); in TEST_F() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dmean_stddev_normalization.cc405 Node* mul_node; in TryMeanStdDevNormalization() local
/external/pytorch/torch/csrc/jit/tensorexpr/
Danalysis.h388 auto mul_node = to<Mul>(v->value()); in visit() local
Dexpr.cpp523 auto mul_node = to<Mul>(cur_stride); in is_cont_with() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dsplit_utils_test.cc287 const NodeDef& mul_node = GetNode("mul", second); in TEST() local
/external/ComputeLibrary/src/graph/
DGraphBuilder.cpp703 …NodeID mul_node = GraphBuilder::add_elementwise_node(g, params, input, mul_const_nidxp, … in add_scale_layer() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc829 const NodeDef* mul_node = node_map.GetNode(HoistMulName("Add_6")); in TEST_F() local