Home
last modified time | relevance | path

Searched defs:axis_node (Results 1 – 25 of 25) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tf/
Dtf_argmax_parser.cc34 auto axis_node = tf_node_map.at(tf_axis_input_name); in Parse() local
Dtf_argmin_parser.cc37 auto axis_node = tf_node_map.at(tf_axis_input_name); in Parse() local
Dtf_concat_parser.cc33 auto axis_node = GetConstInputNode(tf_node_map, tf_op.input(tf_op.input_size() - 1)); in Parse() local
Dtf_gather_parser.cc43 auto axis_node = GetConstInputNode(tf_node_map, tf_op.input(THIRD_INPUT)); in Parse() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dalltoall_fusion.cc76 ValueNodePtr axis_node = std::make_shared<ValueNode>(MakeValue<int64_t>(split_dim)); in CreateSplitNode() local
160 ValueNodePtr axis_node = std::make_shared<ValueNode>(MakeValue<int64_t>(concat_dim)); in CreateConcatNode() local
Dconcat_outputs_for_all_gather.cc87 auto axis_node = concat_cnode->input(data_input_num); in GenerateKernelBuildInfo() local
114 …auto axis_node = opt::CreateValueNodeWithKernelInfo(func_graph, MakeValue(static_cast<int64_t>(0))… in InsertConcatForOutput() local
Dclip_by_norm_fission.cc136 auto axis_node = AnfAlgo::ConvertValueToNode(kernel_graph, MakeValue<std::vector<int64_t>>(axis)); in CreateReduceSumNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/
Donehot_mapper.cc54 auto axis_node = NewValueNode(MakeValue<int64_t>(axis)); in AddAttrAxisToLastInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/add_attr_to_node/
Dsplit_fusion.cc38 auto axis_node = common::AnfAlgo::GetInputNode(cnode, axis_idx); in SplitFusionProcess() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/optimizer/
Dsoftmax_grad_fusion.cc43 auto axis_node = common::AnfAlgo::GetInputNode(reduce_sum_node, 1); in NeedFusion() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dfloatstatus_fusion.cc55 auto axis_node = opt::GetAnfNodeByVar(equiv, axis_); in Process() local
Daxis_normalizer.cc100 auto axis_node = cnode->input(axis_idx); in Process() local
Dgraph_kernel_helper.cc234 auto axis_node = cnode->input(kIndex2)->cast<ValueNodePtr>(); in GetReduceAxis() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dmaketuple_unify_mindir.cc167 auto axis_node = CreateValueNode(func_graph, MakeValue(axis)); in CreateConcat() local
Dlamb_fission.cc184 auto axis_node = CreateValueNode(graph, MakeValue(axis)); in CreateLayerNormNode() local
Dconvert_pad_v3_paddings.cc114 auto axis_node = CreateValueNodeWithKernelInfo(func_graph, MakeValue(axis)); in CreateConcatNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dclip_by_norm_fission.cc117 auto axis_node = CreateValueNode(graph, MakeValue(std::make_shared<tensor::Tensor>(range))); in GetAxisNode() local
153 auto axis_node = GetAxisNode(func_graph, clip_by_norm); in CreateReduceSumNode() local
Dreduce_axis_update.cc126 const auto &axis_node = cnode->input(kAxisIndex); in IsAxisEmptySequence() local
Dsparse_softmax_cross_entropy_with_logits_unify_mindir.cc220 …auto axis_node = CreateTensorInput(kernel_graph, CreateValueNode(MakeValue(range), kNumberTypeInt6… in GetAxisNode() local
241 auto axis_node = GetAxisNode(graph, softmax_output_node); in CreateReduceMean() local
285 auto axis_node = kernel_graph->NewValueNode(axis_abstract, MakeValue(axis)); in CreateExpandDims() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dadjust_resize_dims_pass.cc189 …auto axis_node = BuildIntVecParameterNode(func_graph, {kAxis_0}, shape_node->fullname_with_scope()… in CreateGatherCNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/function/
Dfunc_pass.cc201 auto axis_node = func_builder_->NewFuncNode(axis_v, nullptr, InputType::kConstant); in CreateExpandDims() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc1246 auto axis_node = BuildIntVecParameterNode(func_graph, axis, cnode_name + "_axis"); in GenGatherNode() local
1275 auto axis_node = BuildIntVecParameterNode(func_graph, axis, cnode_name + "_axis"); in GenGatherNodeDynamicIndex() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/operator/composite/
Dtensor_index.cc765 auto axis_node = NewValueNode(MakeValue(static_cast<int64_t>(-1))); in RemakeTupleIndex() local
1138 auto axis_node = NewValueNode(MakeValue(static_cast<int64_t>(-1))); in SetItemByTuple() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/
Dir_pass.cc372 auto axis_node = PyNativeAlgo::Common::CreateValueNodeByValue(MakeValue(axis), axis_abstract); in CreateExpandDims() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/
Dop_node.cc895 auto axis_node = inputs.back(); in CalcConcat() local