Home
last modified time | relevance | path

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

123456789

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/
Donnx_pad_adjust.cc32 CNodePtr NewReshapeOpNode(const FuncGraphPtr &func_graph, const AnfNodePtr &input_node, const std::… in NewReshapeOpNode()
62 bool AdjstVariablePadding(const FuncGraphPtr &func_graph, const AnfNodePtr &input_node) { in AdjstVariablePadding()
98 bool AdjstConstPadding(const CNodePtr &cnode, const AnfNodePtr &input_node) { in AdjstConstPadding()
137 auto input_node = cnode->input(2); in Adjust() local
Donnx_deform_conv2d_adjust.cc28 CNodePtr NewReshapeOpNode(const FuncGraphPtr &func_graph, const AnfNodePtr &input_node, const AnfNo… in NewReshapeOpNode()
46 CNodePtr NewShapeOpNode(const FuncGraphPtr &func_graph, const AnfNodePtr &input_node) { in NewShapeOpNode()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/
Dconv1d_inout_adjust.cc42 … Conv1DInOutAdjust::NewUnsqueezeOpNode(const FuncGraphPtr &func_graph, const AnfNodePtr input_node, in NewUnsqueezeOpNode()
61 …tr Conv1DInOutAdjust::NewSqueezeOpNode(const FuncGraphPtr &func_graph, const AnfNodePtr input_node, in NewSqueezeOpNode()
108 CNodePtr Conv1DInOutAdjust::getOutputNode(const FuncGraphPtr &func_graph, const CNodePtr input_node, in getOutputNode()
159 auto input_node = cnode->input(1); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/quant_helper/
Dquant_node_pass.cc105 int QuantNodePass::CheckNodeDType(const CNodePtr &cnode, const AnfNodePtr &input_node, size_t input… in CheckNodeDType()
132 int QuantNodePass::DoParameterNodeQuant(const CNodePtr &cnode, const ParameterPtr &input_node, size… in DoParameterNodeQuant()
179 int QuantNodePass::DoValueNodeQuant(const CNodePtr &cnode, const ValueNodePtr &input_node, size_t i… in DoValueNodeQuant()
225 auto input_node = cnode->input(i); in DoFullQuant() local
263 bool QuantNodePass::CanTensorQuantized(const CNodePtr &cnode, const AnfNodePtr &input_node) { in CanTensorQuantized()
Dtransform_uint8_pass.cc73 int TransformUint8Pass::DoParameterNodeTrans(const CNodePtr &cnode, const ParameterPtr &input_node, in DoParameterNodeTrans()
192 auto input_node = cnode->input(index); in DoNodeDTypeTrans() local
250 auto input_node = cnode->input(index); in CopyQuantParam() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dkvcache_quant_pass.cc56 …VCacheQuantPass::NewQuantNode(const FuncGraphPtr &func_graph, const AnfNodePtr &input_node, TypeId… in NewQuantNode()
81 …acheQuantPass::NewAscendAntiQuantNode(const FuncGraphPtr &func_graph, const AnfNodePtr &input_node, in NewAscendAntiQuantNode()
112 auto input_node = cast_cnode->input(kIndexOne); in ReplaceCastOpToQuantOp() local
137 auto input_node = cast_cnode->input(kIndexOne); in ReplaceCastOpToAntiQuantOp() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/
Dfull_quant_quantizer.cc51 const AnfNodePtr &input_node, const std::unique_ptr<DataDistribution> &info) const { in GetQuantParam() argument
111 int FullQuantQuantizer::IsSupportWeightQuant(const AnfNodePtr &input_node) { in IsSupportWeightQuant()
133 int FullQuantQuantizer::DoParameterNodeQuant(const CNodePtr &cnode, const ParameterPtr &input_node, in DoParameterNodeQuant()
173 int FullQuantQuantizer::DoValueNodeQuant(const CNodePtr &cnode, const ValueNodePtr &input_node, siz… in DoValueNodeQuant()
189 …ullQuantQuantizer::QuantNodeGraphInput(const PrimitivePtr &primitive, const AnfNodePtr &input_node, in QuantNodeGraphInput()
213 int FullQuantQuantizer::QuantNodeCNode(const CNodePtr &cnode, const AnfNodePtr &input_node, in QuantNodeCNode()
236 int FullQuantQuantizer::QuantValueNode(const CNodePtr &cnode, const AnfNodePtr &input_node, size_t … in QuantValueNode()
268 auto input_node = cnode->input(i); in QuantNodeSimpleOp() local
Dinsert_quant_node_manager.cc56 int InsertQuantNodeManager::SetCastNodeAbstract(const CNodePtr &cnode, const AnfNodePtr &input_node, in SetCastNodeAbstract()
77 int InsertQuantNodeManager::CheckDataType(const AnfNodePtr &input_node, TypeId check_type_id) const… in CheckDataType() argument
331 auto input_node = cnode->input(index); in InsertForwardQuantNodeNew() local
415 auto input_node = cnode->input(index); in InsertForwardQuantNode() local
563 auto input_node = cnode->input(index); in InsertForwardCastNode() local
591 auto input_node = cnode->input(index); in InsertCastNodeForFullQuant() local
658 auto input_node = cnode->input(input_index); in InsertQuantDtypeCastFlyNode() local
697const AnfNodePtr input_node, TypeId src_dtype, in CreateQuantInputCastNode()
741 auto input_node = cnode->input(input_index); in CalculateScaleZPNode() local
840 auto input_node = cnode->input(input_index); in InsertAscendAntiQuantNode() local
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dreshape_reshape_fusion.cc46 auto input_node = std::make_shared<Var>(); in DefinePreReshapePattern() local
62 auto input_node = std::make_shared<Var>(); in DefinePostReshapePattern() local
77 auto input_node = std::make_shared<Var>(); in DefineReshapeReshapePattern() local
181 auto input_node = pre_cnode->input(1); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Dflatten_concat_fission.cc36 int64_t GetElemNum(const AnfNodePtr &input_node) { in GetElemNum()
49 CNodePtr NewFlattenNode(const FuncGraphPtr &func_graph, const AnfNodePtr &input_node, int64_t elem_… in NewFlattenNode()
139 auto input_node = common::AnfAlgo::GetInputNode(cnode_ptr, i); in ExpandFlattenConcatTupleInput() local
187 auto input_node = cnode->input(i); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/mapper/
Dbatch_norm_mapper.cc37 auto input_node = cnode->input(i); in SetBnDataInfo() local
76 auto input_node = cnode->input(i); in SetFusedBnDataInfo() local
Dop_mapper.cc170 auto input_node = cnode->input(i); in SetConvFcDataInfo() local
205 auto input_node = cnode->input(i); in SetRecurrentDataInfo() local
247 auto input_node = cnode->input(i); in SetRecurrentOnnxInfo() local
352 auto input_node = cnode->input(i); in PushOfflineArgs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/
Dvalue_depend_op_utils.cc65 auto input_node = inputs[i + 1]; in IsConstInput() local
106 auto input_node = inputs[i + 1]; in AddConstInputToAttr() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/
Dgrouped_pairwise_exchange_alltoall.cc48 auto input_node = marked_node->input(1); in FindFrontAlltoall() local
179 CNodePtr NewSplitNode(const AnfNodePtr &input_node, size_t split_dim, size_t split_num) { in NewSplitNode()
200 CNodePtr NewSplitNode(const AnfNodePtr &input_node, size_t split_dim, size_t split_num, const Shape… in NewSplitNode()
224 CNodePtr NewConcatNode(const AnfNodePtr &input_node, size_t concat_dim, size_t input_num) { in NewConcatNode()
258 CNodePtr NewTupleGetItemNode(const AnfNodePtr &input_node, size_t output_index) { in NewTupleGetItemNode()
272 void MakeSortedSplitGetItemNodes(const AnfNodePtr &input_node, const std::vector<int64_t> &sort_idx, in MakeSortedSplitGetItemNodes()
284 void NewTupleGetItemNodes(const AnfNodePtr &input_node, size_t split_num, std::vector<AnfNodePtr> *… in NewTupleGetItemNodes()
292 CNodePtr NewNeighborExchangeNode(const AnfNodePtr &input_node, const std::vector<int64_t> &send_ran… in NewNeighborExchangeNode()
334 void CreateNeighborExchangeNodes(const AnfNodePtr &input_node, size_t split_dim, size_t concat_dim, in CreateNeighborExchangeNodes()
415 void CloneScaledGraph(const std::vector<CNodePtr> &old_cnodes, const AnfNodePtr &input_node, size_t… in CloneScaledGraph()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dconvert_resize_nearest_neighbor_x_dtype.cc69 auto input_node = cnode->input(input_idx); in ConvertDataTypeForCNodeInput() local
98 auto input_node = cnode->input(kIndex2); in Process() local
Dall_to_all_v_for_ge.cc147 [&graph](const auto &input_node) { in CreateFlattenReshapeNodes()
165 for (const auto &input_node : input_nodes) { in CreateConcatNode() local
184 CNodePtr CreateAllToAllvForGENode(const FuncGraphPtr &graph, const AnfNodePtr &input_node, const CN… in CreateAllToAllvForGENode()
222 CNodePtr CreateSplitNode(const FuncGraphPtr &graph, const AnfNodePtr &input_node, in CreateSplitNode()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/add_attr_to_node/
Daddn_fusion.cc30 auto input_node = cnode->input(idx); in AddNFusionProcess() local
Dconcat_offset_v1_fusion.cc32 auto input_node = cnode->input(idx); in ConcatOffsetV1FusionProcess() local
Daccumulate_n_v2_fusion.cc32 auto input_node = cnode->input(idx); in AccumulateNV2FusionProcess() local
Dparallel_concat_fusion.cc31 auto input_node = cnode->input(idx); in ParallelConcatFusionProcess() local
Dsparse_concat_fusion.cc34 auto input_node = cnode->input(idx); in SparseConcatFusionProcess() local
Dsparse_cross_fusion.cc34 auto input_node = cnode->input(idx); in SparseCrossFusionProcess() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/trt_pass/
Dgraph_partitioner.cc146 auto input_node = common::AnfAlgo::GetInputNode(node->cast<CNodePtr>(), input_index); in ExistCycleAfterMerge() local
170 auto input_node = common::AnfAlgo::GetInputNode(node->cast<CNodePtr>(), input_index); in MergeParentBranchRecursively() local
199 auto input_node = common::AnfAlgo::GetInputNode(node->cast<CNodePtr>(), input_index); in NodeGrouping() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fisson/
Diter_node_outputs.cc33 for (const auto &input_node : inputs) { in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dremove_reshape_pair.cc54 auto input_node = common::AnfAlgo::GetInputNode(in_reshape, 0); in Process() local

123456789