| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | insert_tensor_move_for_hccl_op.cc | 80 auto real_input = kernel_with_index.first; in NeedInsertTensorMove() local
82 if (IsParameterOrValueNode(real_input)) { in NeedInsertTensorMove()
86 if (kernel_query_->IsTbeRef(real_input)) { in NeedInsertTensorMove()
91 …if (kNeedInsertTensorMoveOpSet.find(AnfAlgo::GetCNodeName(real_input)) != kNeedInsertTensorMoveOpS… in NeedInsertTensorMove()
104 if (opt::IsNopNode(real_input)) { in NeedInsertTensorMove()
105 auto cnode = real_input->cast<CNodePtr>(); in NeedInsertTensorMove()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | convert_tuple_output_to_maketuple.cc | 68 auto real_input = AnfAlgo::GetTupleGetItemRealInput(cnode); in Process() local
69 MS_EXCEPTION_IF_NULL(real_input); in Process()
70 if (!real_input->isa<Parameter>() && !real_input->isa<ValueNode>()) { in Process()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | switch_actor.cc | 238 const auto &real_input = AnfAlgo::VisitKernelWithReturnType(node, 0); in AddInput() local
240 if (AnfAlgo::CheckPrimitiveType(real_input.first, prim::kPrimMakeTuple)) { in AddInput()
241 const auto &inputs = real_input.first->cast<CNodePtr>()->inputs(); in AddInput()
245 } else if (IsCallNode(real_input.first)) { in AddInput()
247 const auto call_output_num = FetchOutputSizebyCallNode(real_input.first, &call_nodes); in AddInput()
249 …XCEPTION) << "Invalid output num for call input:" << AnfAlgo::GetNodeDebugString(real_input.first); in AddInput()
252 AddInput({real_input.first, i}, branch); in AddInput()
254 …} else if (real_input.first->isa<ValueNode>() && real_input.first->cast<ValueNodePtr>()->value()->… in AddInput()
255 const auto &value = real_input.first->cast<ValueNodePtr>()->value(); in AddInput()
258 AddInput({real_input.first, i}, branch); in AddInput()
[all …]
|
| /third_party/mindspore/mindspore/lite/src/c_api/ |
| D | model_c.cc | 142 auto real_input = model_inputs[i]; in Predict() local
144 if (user_input->DataType() != static_cast<DataType>(real_input->data_type())) { in Predict()
147 << ", real:" << real_input->tensor_name(); in Predict()
155 old_data.push_back(real_input->data()); in Predict()
156 if (real_input->data_type() == kObjectTypeString) { in Predict()
160 real_input->set_shape(shape); in Predict()
161 real_input->set_data(user_input->MutableData()); in Predict()
163 if (user_input->MutableData() != real_input->data()) { in Predict()
164 if (real_input->Size() != user_input->DataSize()) { in Predict()
169 real_input->set_data(user_input->MutableData()); in Predict()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | ub_pattern_fusion.cc | 116 auto real_input = AnfAlgo::VisitKernel(input, 0); in CreateFusionOpKernelInfo() local
117 (void)inputs_format.emplace_back(AnfAlgo::GetOutputFormat(real_input.first, real_input.second)); in CreateFusionOpKernelInfo()
118 …nputs_data_type.emplace_back(AnfAlgo::GetOutputDeviceDataType(real_input.first, real_input.second)… in CreateFusionOpKernelInfo()
238 auto real_input = AnfAlgo::VisitKernel(cnode->input(idx), 0); in GetFusionScopeInputNodeList() local
239 … if (std::find(fusion_info.anf_nodes.begin(), fusion_info.anf_nodes.end(), real_input.first) == in GetFusionScopeInputNodeList()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.cc | 119 auto real_input = node_with_index.first; in GetTransInputNodePtr() local
120 if (real_input->isa<ValueNode>() || real_input->isa<Parameter>()) { in GetTransInputNodePtr()
131 if (real_input->isa<Parameter>()) { in GetTransInputNodePtr()
132 SetGroupAttr(real_input->cast<ParameterPtr>(), input_node, transdata, dest_format); in GetTransInputNodePtr()
|
| /third_party/mindspore/tests/ut/cpp/tbe/ |
| D | tbe_json_creator_test.cc | 296 auto real_input = AnfAlgo::VisitKernel(cnode->input(idx), 0); in TEST_F() local
297 …if (std::find(compute_nodes.begin(), compute_nodes.end(), real_input.first) == compute_nodes.end()… in TEST_F()
355 auto real_input = AnfAlgo::VisitKernel(cnode->input(idx), 0); in TEST_F() local
356 …if (std::find(compute_nodes.begin(), compute_nodes.end(), real_input.first) == compute_nodes.end()… in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.cc | 845 for (auto &real_input : real_inputs) { in ProcessNodeRetFunc() local
846 auto new_parameter = kernel_graph->NewParameter(real_input->abstract()); in ProcessNodeRetFunc()
1443 auto real_input = kernel_with_index.first; in GetOpInputTensors() local
1444 MS_EXCEPTION_IF_NULL(real_input); in GetOpInputTensors()
1446 if (real_input->isa<ValueNode>()) { in GetOpInputTensors()
1447 tensor = GetValueNodeOutputTensor(real_input, kernel_with_index.second); in GetOpInputTensors()
1448 } else if (real_input->isa<Parameter>()) { in GetOpInputTensors()
1449 tensor = GetParameterOutputTensor(real_input, parameter_index, graph_inputs); in GetOpInputTensors()
1450 } else if (real_input->isa<CNode>()) { in GetOpInputTensors()
1452 if (AnfAlgo::IsControlOpExecInBackend(real_input)) { in GetOpInputTensors()
[all …]
|
| D | ascend_session.cc | 1090 auto real_input = kernel_with_index.first; in GetOpInputStubTensors() local
1091 MS_EXCEPTION_IF_NULL(real_input); in GetOpInputStubTensors()
1093 if (real_input->isa<ValueNode>()) { in GetOpInputStubTensors()
1094 tensor = GetValueNodeOutputTensor(real_input, kernel_with_index.second); in GetOpInputStubTensors()
1096 } else if (real_input->isa<Parameter>()) { in GetOpInputStubTensors()
1097 tensor = GetParameterOutputTensor(real_input, parameter_index, graph_inputs); in GetOpInputStubTensors()
1098 auto parameter = real_input->cast<ParameterPtr>(); in GetOpInputStubTensors()
1102 } else if (real_input->isa<CNode>()) { in GetOpInputStubTensors()
1108 MS_LOG(EXCEPTION) << "Invalid input node, node = " << real_input->DebugString(); in GetOpInputStubTensors()
1112 << real_input->fullname_with_scope() << "-" << kernel_with_index.second; in GetOpInputStubTensors()
|
| D | anf_runtime_algorithm.cc | 122 auto real_input = AnfAlgo::GetInputNode(make_tuple, index); in GetRealOutputRecursively() local
123 return GetRealOutputRecursively(real_input, 0, inputs); in GetRealOutputRecursively()
2230 auto real_input = input_with_index.first; in InferShape() local
2233 MS_EXCEPTION_IF_NULL(real_input); in InferShape()
2241 auto real_abs = real_input->abstract(); in InferShape()
2243 real_input->abstract()->set_value(tensor_ptr); in InferShape()
2254 auto base_shape = real_input->Shape(); in InferShape()
2259 auto abs = real_input->abstract()->cast<abstract::AbstractTuplePtr>(); in InferShape()
2267 (void)args_spec_list.emplace_back(real_input->abstract()); in InferShape()
2360 auto real_input = GetRealInputIndex(kernel, idx); in CacheAddrForGraph() local
[all …]
|
| D | kernel_graph.cc | 80 auto real_input = child_graph->output(); in GetCallRealOutputs() local
81 auto child_real_inputs = GetCallRealOutputs(real_input); in GetCallRealOutputs()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | backend.cc | 525 auto real_input = kernel_with_index.first; in GetControlOpInput() local
526 MS_EXCEPTION_IF_NULL(real_input); in GetControlOpInput()
528 if (!real_input->isa<ValueNode>()) { in GetControlOpInput()
538 const auto &value_node = real_input->cast<ValueNodePtr>(); in GetControlOpInput()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 1149 auto real_input = AnfAlgo::GetRealInputIndex(kernel, i); in GenLaunchArgs() local
1150 auto device_address = AnfAlgo::GetPrevNodeOutputAddr(kernel, real_input, visit_nop_node); in GenLaunchArgs()
1314 auto real_input = AnfAlgo::GetRealInputIndex(kernel, j); in AssignKernelAddress() local
1315 auto kernel_with_index = AnfAlgo::GetPrevNodeOutput(kernel, real_input, true); in AssignKernelAddress()
1511 auto real_input = AnfAlgo::GetRealInputIndex(kernel, idx); in LaunchKernelMod() local
1512 auto device_address = AnfAlgo::GetPrevNodeMutableOutputAddr(kernel, real_input); in LaunchKernelMod()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_stream_assign.cc | 1378 auto real_input = AnfAlgo::VisitKernel(inputs[i], 0); in GetInputKernels() local
1379 auto node = real_input.first; in GetInputKernels()
1687 auto real_input = AnfAlgo::VisitKernel(input, 0); in GetIndependentMaxTarget() local
1688 if (key == real_input.first.get()) { in GetIndependentMaxTarget()
2002 auto real_input = AnfAlgo::VisitKernel(input, 0); in FindTargetOp() local
2003 if (node == real_input.first) { in FindTargetOp()
|