Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A 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
H A Dfold_batch_norms.cc52 const NodeDef& mul_node = match.node; in FoldBatchNorms() local
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
H A Dfuse_mul_to_conv_test.cc64 auto mul_node = graph.NewNode(); in TEST() local
115 auto mul_node = graph.NewNode(); in TEST() local
H A Dfuse_mul_to_conv.cc49 auto& mul_node = *sequence[1]; in ApplyToNodesSequence() local
113 auto& mul_node = *sequence[0]; in ApplyToNodesSequence() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
H A Dconv_pointwise.cc158 NodeContext mean_node, mul_node, slice_node; in GetOffset() local
221 NodeContext mul_node; in TryFusedPointwiseConv() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dmean_stddev_normalization.cc405 Node* mul_node; in TryMeanStdDevNormalization() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Danalysis.h388 auto mul_node = to<Mul>(v->value()); in visit() local
H A Dexpr.cpp523 auto mul_node = to<Mul>(cur_stride); in is_cont_with() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dsplit_utils_test.cc287 const NodeDef& mul_node = GetNode("mul", second); in TEST() local
/aosp_15_r20/external/ComputeLibrary/src/graph/
H A DGraphBuilder.cpp703 …NodeID mul_node = GraphBuilder::add_elementwise_node(g, params, input, mul_const_nidxp, … in add_scale_layer() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer_test.cc829 const NodeDef* mul_node = node_map.GetNode(HoistMulName("Add_6")); in TEST_F() local