Home
last modified time | relevance | path

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

12345

/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_pad_adjust.cc57 CNodePtr NewReshapeOpNode(const FuncGraphPtr &func_graph, const AnfNodePtr &input_node, const std::… in NewReshapeOpNode()
78 CNodePtr NewTransposeOpNode(const FuncGraphPtr &func_graph, const AnfNodePtr &input_node, in NewTransposeOpNode()
110 auto input_node = cnode->input(2); in Adjust() local
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dquant_cast.cc49 auto input_node = cnode->input(input_index); in InsertCastNode() local
110 STATUS CheckDataType(const AnfNodePtr &input_node, bool is_graph_input) { in CheckDataType()
135 auto input_node = cnode->input(i); in Run() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dconv1d_inout_adjust.cc35 … Conv1DInOutAdjust::NewUnsqueezeOpNode(const FuncGraphPtr &func_graph, const AnfNodePtr input_node, in NewUnsqueezeOpNode()
52 …tr Conv1DInOutAdjust::NewSqueezeOpNode(const FuncGraphPtr &func_graph, const AnfNodePtr input_node, in NewSqueezeOpNode()
128 auto input_node = cnode->input(1); in Run() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dconvert_const_input_to_tensor_input.cc31 AnfNodePtr CreateTensorInput(const KernelGraphPtr &kernel_graph, const AnfNodePtr &input_node) { in CreateTensorInput()
77 auto input_node = inputs[i + 1]; in ConstInputToTensorInput() local
Dconvert_const_scalar_to_tensor.cc27 AnfNodePtr CreateTensorInput(const KernelGraphPtr &kernel_graph, const AnfNodePtr &input_node) { in CreateTensorInput()
/third_party/mindspore/mindspore/ccsrc/runtime/device/executor/
Ddynamic_kernel.cc101 auto input_node = input_node_with_index.first; in InferShapeRecursive() local
107 void DynamicKernel::InferShapeForNopNode(AnfNodePtr *input_node) { in InferShapeForNopNode()
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Diter_node_outputs.cc32 for (const auto &input_node : inputs) { in Run() local
Dnode_out_shapes.cc34 for (const auto &input_node : cnode->inputs()) { in Run() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Ddata_prepare_actor.cc210 const auto &input_node = input_nodes[j]; in PrepareDataForDeviceTensorStore() local
241 const auto &input_node = input_nodes[j]; in PrepareDataForHostTensorQueue() local
285 const auto &input_node = input_nodes[j]; in PrepareDataForStepMode() local
520 const auto &input_node = control_node_parameters[i]; in PrepareDeviceTensorStoreForControlNode() local
546 const auto &input_node = control_node_parameters[i]; in PrepareHostTensorQueueForControlNode() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dgraph_partitioner.cc140 auto input_node = AnfAlgo::GetInputNode(node->cast<CNodePtr>(), input_index); in ExistCycleAfterMerge() local
164 auto input_node = AnfAlgo::GetInputNode(node->cast<CNodePtr>(), input_index); in MergeParentBranchRecursively() local
193 auto input_node = AnfAlgo::GetInputNode(node->cast<CNodePtr>(), input_index); in NodeGrouping() local
Dtrt_converter_context.cc100 for (auto input_node : graph_inputs) { in InitInputTable() local
223 for (const auto &input_node : func_graph_->parameters()) { in GetGraphInputs() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dremove_reshape_pair.cc53 auto input_node = AnfAlgo::GetInputNode(in_reshape, 0); in Process() local
Dmul_addn_fusion.cc86 auto input_node = mul->input(index); in Process() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dremove_format_transform_pair.cc41 auto input_node = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process() local
Dremove_redundant_format_transform.cc39 auto input_node = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dspecify_graph_input_format.cc53 auto input_node = input->cast<ParameterPtr>(); in HandleGraphInput() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Ddynamic_rnn_grad_reformat.cc40 auto input_node = input_node_with_idx.first; in Process() local
Dinsert_transdata_for_runop.cc41 auto input_node = AnfAlgo::GetInputNode(cnode, index); in Run() local
Dremove_internal_output.cc73 auto input_node = cnode->input(1); in Process() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/reporter/
Dgraph_desc_reporter.cc37 auto input_node = input_node_with_index.first; in ReportData() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dadd_placeholder_for_dynamic_rnn.cc55 auto input_node = AnfAlgo::GetInputNode(cnode, in_idx); in Process() local
Dadd_placeholder_for_dynamic_gru.cc64 auto input_node = AnfAlgo::GetInputNode(cnode, real_input_index); in Process() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dgelu_fusion.cc39 auto input_node = utils::cast<AnfNodePtr>((*equiv)[input_]); in CreateGeLUNode() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_helper.cc41 AnfNodePtr CreateReshapeNode(const FuncGraphPtr &func_graph, const AnfNodePtr &input_node, in CreateReshapeNode()
112 auto input_node = AnfAlgo::GetInputNode(node, index); in GetTransInputNodePtr() local
199 …AnfNodePtr input_node = is_insert_input ? AnfAlgo::GetInputNode(node->cast<CNodePtr>(), insert_ind… in AddTransOpNodeToGraph() local
396 AnfNodePtr input_node = GetTransInputNodePtr(func_graph, cnode, input_index, kernel_select); in InsertTransOpForInput() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dtransdata_split.cc64 auto input_node = cnode->input(kIndex1); in DoSplit() local

12345