Home
last modified time | relevance | path

Searched refs:set_inputs (Results 1 – 25 of 46) sorted by relevance

12

/third_party/mindspore/mindspore/lite/src/delegate/npu/pass/
Dnpu_pass_utils.cc54 op->set_inputs(in_tensors); in UpdateOp()
96 post_op->set_inputs(cur_in_tensors); in UpdateNH2NCTransNodePostOp()
110 post_op->set_inputs(post_in_tensors); in UpdateNC2NHPostOpInTensors()
129 post_op->set_inputs(post_in_tensors); in UpdateNC2NHTransNodePostOp()
Dnpu_fusion_pass.cc166 cur_op->set_inputs(tensors_vec); in UpdatePreTensors()
194 post_post_op->set_inputs({nc_tensor}); in UpdatePostTensors()
219 post_op->set_inputs(tensors_vec); in UpdatePostTensors()
400 post_op->set_inputs(tensors_vec); in FormatFusion()
434 pre_out_ops[index]->set_inputs(tensors_vec); in FormatFusion()
Dnpu_transform_pass.cc153 non_insert_op->set_inputs(inputs); in InsertPostNodes()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dinfershape_pass.cc33 cnode->set_inputs(origin_inputs); in GetCNodeCertainInputFormat()
37 cnode->set_inputs(origin_inputs); in GetCNodeCertainInputFormat()
45 cnode->set_inputs(origin_inputs); in GetCNodeCertainInputFormat()
328 return_node->set_inputs(origin_input); in SetSubGraphOutput()
366 return_node->set_inputs(origin_inputs); in SetSubGraphAbstract()
Dnode_infershape.cc214 cnode->set_inputs(specify_inputs); in GetIntVecInput()
217 cnode->set_inputs(origin_inputs); in GetIntVecInput()
220 cnode->set_inputs(origin_inputs); in GetIntVecInput()
252 cnode->set_inputs(origin_inputs); in GetCNodeInputTensors()
259 cnode->set_inputs(origin_inputs); in GetCNodeInputTensors()
282 cnode->set_inputs(origin_inputs); in GetCNodeInputTensors()
Ddecrease_transpose_algo.cc77 cur_node->set_inputs(origin_inputs); in FindAreaSurroundedByTranspose()
320 cnode->set_inputs(origin_inputs); in InsertPreTransNode()
323 cnode->set_inputs(origin_inputs); in InsertPreTransNode()
576 return_node->set_inputs(origin_input); in SetSubGraphOutput()
614 return_node->set_inputs(origin_inputs); in SetSubGraphAbstract()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_label_assign.cc41 node->set_inputs({node->input(0)}); in UpdateLabelGoto()
64 node->set_inputs({node->input(kAnfPrimitiveIndex), node->input(kFirstDataInputIndex)}); in UpdateLabelSwitch()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Doptimize_dependence.cc48 new_depend->set_inputs(new_depend_inputs); in CreateNewDependNode()
143 new_make_tuple->set_inputs(new_make_tuple_inputs); in ReplaceMakeTuple()
Dconvert_tuple_input_to_dynamic_input.cc83 cnode_ptr->set_inputs(plant_inputs); in ConvertMakeTupleInputToPlantInputs()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dadd_placeholder_for_dynamic_rnn.cc74 new_node->set_inputs(new_inputs); in Process()
Dadd_placeholder_for_dynamic_gru.cc77 new_node->set_inputs(new_inputs); in Process()
Dinsert_pad_for_nms_with_mask.cc87 new_node->set_inputs(new_inputs); in Process()
Dinsert_tensor_move_for_cascade.cc104 new_hccl_node->set_inputs(new_inputs); in InsertTensorMove()
Dinsert_tensor_move_for_hccl_op.cc136 new_hccl_node->set_inputs(new_inputs); in InsertTensorMove()
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_inputs_adjust.cc298 cnode->set_inputs(inputs); in AdjustStridedSlice()
317 cnode->set_inputs(new_input); in AdjustResize()
356 cnode->set_inputs(new_resize_inputs); in AdjustResize()
/third_party/mindspore/include/api/
Dkernel.h60 …virtual void set_inputs(const std::vector<mindspore::MSTensor> &in_tensors) { this->inputs_ = in_t… in set_inputs() function
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/
Dbatch_norm_bert_fission_test.cc59 bn_cnode->set_inputs(new_inputs); in TEST_F()
/third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/
Dspatial_node_adapter.cc128 cnode->set_inputs(replace_node); in AdapteNodeWithDynamicInput()
Dprimitive_mapper.cc196 cnode->set_inputs(inputs); in AddAttrToInput()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dadd_stitch_atomic_clean_gpu.cc57 composite_node->cast<CNodePtr>()->set_inputs(inputs); in ProcessOriginCNode()
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dformat_utils.cc171 cnode->set_inputs(inputs); in RemoveIfMonad()
/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Dnpu_op.h81 …void set_inputs(const std::vector<mindspore::MSTensor> &in_tensors) { this->inputs_ = in_tensors; } in set_inputs() function
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_inputs_adjust.cc86 cnode->set_inputs(new_inputs); in ReorderCnodeInputs()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dprogram_specialize.cc190 c_new_node->set_inputs(new_inputs); in UpdateNewCNodeInputs()
353 c_new->set_inputs(new_inputs); in ProcessNode()
731 new_node->set_inputs(new_inputs); in ProcessCNode()
/third_party/mindspore/mindspore/lite/tools/anf_exporter/
Dfetch_content.cc455 cnode->set_inputs(inputs); in RemoveIfDepend()
489 cnode->set_inputs(inputs); in RemoveIfMakeTuple()

12