Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/
Dcallback_impl.cc37 …KernelWithIndex kernel_with_index = AnfUtils::VisitKernel(node->cast<CNodePtr>()->input(i + 1), 0); in GetInputInferShape() local
79 …KernelWithIndex kernel_with_index = AnfUtils::VisitKernel(node->cast<CNodePtr>()->input(i + 1), 0); in GetInputInferType() local
112 auto kernel_with_index = AnfUtils::VisitKernel(cnode->input(i + 1), 0); in GetInputFormat() local
162 auto kernel_with_index = AnfUtils::VisitKernel(outputs[i], 0); in SetGraphKernelNodeKernelInfo() local
Dformat_recognition.cc92 auto kernel_with_index = AnfUtils::VisitKernel(cnode->input(i), 0); in SetOutputsFormat() local
Dgraph_kernel_expander_lite.cc79 auto kernel_with_index = AnfUtils::VisitKernel(cnode->input(i), 0); in GetFixedFormat() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/
Dcpu_simple_mem_plan.cc31 auto kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(kernel, i); in MemPlan() local
75 auto kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(kernel, i); in MemAssign() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dinsert_identity.cc90 auto kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(cnode, index, false); in InsertTransIdentityForInput() local
112 auto kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(cnode, index, false); in InsertCastIdentityForInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dtsa_atomic_add_to_first_tensor.cc108 auto kernel_with_index = common::AnfAlgo::VisitKernel(tsa_first_input.first, 0); in GetOrCreateNewTsaFirstNode() local
164 auto kernel_with_index = common::AnfAlgo::VisitKernel(modified_input, 0); in ChangeKernelBuildInfo() local
Dadd_stitch_atomic_clean_gpu.cc48 auto kernel_with_index = common::AnfAlgo::VisitKernel(inplace_infos[0].second, 0); in CorrectKernelBuildInfo() local
Dinplace_assign_builder.cc82 auto kernel_with_index = common::AnfAlgo::VisitKernel(new_input, 0); in CorrectKernelBuildInfo() local
Dgraph_kernel_helper.cc309 auto kernel_with_index = common::AnfAlgo::VisitKernel(inputs[i], 0); in CreateCNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dsession_basic.cc545 auto kernel_with_index = common::AnfAlgo::VisitKernel(node, 0); in CalculateRefCount() local
574 auto kernel_with_index = common::AnfAlgo::VisitKernel(input, 0); in GetForwardOpOutputRefCount() local
628 for (const auto &kernel_with_index : input_kernel) { in HandleOpInputs() local
673 auto kernel_with_index = make_pair(kernel, out_index++); in HandleOpOutputs() local
774 …const KernelWithIndex &kernel_with_index, const std::map<KernelWithIndex, tensor::BaseTensorPtr> &… in GetCNodeOutputTensor() argument
833 auto kernel_with_index = common::AnfAlgo::VisitKernel(input, 0); in GetOpInputTensors() local
910 …d::function<ValuePtr(const KernelWithIndex &)> fn = [&](const KernelWithIndex &kernel_with_index) … in GetOpInputTensorsFromCNode()
944 KernelWithIndex kernel_with_index; in GetOpInputTensorsFromCNode() local
971 auto kernel_with_index = common::AnfAlgo::VisitKernel(input, 0); in GetOpInputTensorByIndex() local
Danf_runtime_algorithm.cc513 KernelWithIndex kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(anf_node, input_idx); in GetPrevNodeOutputFormat() local
518 KernelWithIndex kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(node, input_idx); in GetPrevNodeOutputReshapeType() local
815 KernelWithIndex kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(anf_node, input_idx); in GetPrevNodeOutputDeviceDataType() local
873 auto kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(cnode, 0); in OutputAddrExist() local
898 KernelWithIndex kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(anf_node, input_idx); in GetPrevNodeOutputAddr() local
904 KernelWithIndex kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(anf_node, input_idx); in GetPrevNodeMutableOutputAddr() local
1018 KernelWithIndex kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(node, input_idx, false); in GetPrevNodeOutputKernelTensor() local
1024 KernelWithIndex kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(node, input_idx, false); in GetOrCreatePrevNodeOutputKernelTensor() local
2003 KernelWithIndex kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(node, input_idx); in GetPrevNodeOutputDetailShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/adapter/
Dcallback_impl.cc53 void GetTypeAndFormats(const device::KernelWithIndex &kernel_with_index, std::vector<TypeId> *input… in GetTypeAndFormats()
136 auto kernel_with_index = AnfUtils::VisitKernel(node, 0); in CollectInputTypesAndFormats() local
182 auto kernel_with_index = common::AnfAlgo::VisitKernel(outputs[i], 0); in SetGraphKernelNodeKernelInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/
Ddeal_ref_output.cc88 session::KernelWithIndex kernel_with_index = common::AnfAlgo::VisitKernel(node, 0); in FindRefOriginNode() local
120 session::KernelWithIndex kernel_with_index = in AddRefNodePairToKernelGraph() local
Dinsert_cast.cc149 auto kernel_with_index = common::AnfAlgo::VisitKernelWithReturnType(cur_input, 0); in InsertCastForInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/
Dinsert_tensor_move_for_hccl_op.cc73 auto kernel_with_index = common::AnfAlgo::VisitKernelWithReturnType(input, 0, true); in NeedInsertTensorMoveForSpecialCase() local
Dconcat_outputs_for_all_gather.cc59 auto kernel_with_index = common::AnfAlgo::VisitKernel(node, 0); in GetNodesOutputInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/
Dacl_helper.cc101 auto kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(node, 0); in SetParameterFormat() local
189 auto kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(node, i); in GetInputBuildInfo() local
511 auto kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(node, i); in GetValidKernelBuildInfo() local
Dop_api_util.cc116 auto kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(node, i); in GetValidKernelBuildInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/hcom/
Dinsert_tensor_move_for_hccl_op_ge.cc129 auto kernel_with_index = common::AnfAlgo::VisitKernelWithReturnType(input, 0, true); in NeedInsertTensorMoveForSpecialCase() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Danfalgo.cc715 [](auto kernel_with_index) { in GetRealPrevNodesOutputInferDataType()
819 KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(node, input_idx); in GetPrevNodeOutputInferShape() local
980 KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(node, input_idx); in GetPrevNodeOutputInferDataType() local
985 KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(node, input_idx); in GetPrevNodeOutputInferType() local
1117 AnfNodePtr AnfAlgo::GetOutputOfGraphkernel(const KernelWithIndex &kernel_with_index) { in GetOutputOfGraphkernel()
1525 auto kernel_with_index = VisitKernel(input_node, 0); in GetPrevNodeOutputPrecision() local
1954 auto kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(node, index); in IsNoneInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/
Dop_runtime_info.cc81 … session::KernelWithIndex kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(node, i, true); in CacheForExecutionOrder() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/data_dump/
Dcpu_e2e_dump.cc105 auto kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(node, j); in DumpInputImpl() local
De2e_dump.cc311 auto kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(node, j); in DumpInputImpl() local
364 auto kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(node, j); in DumpInputData() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/mem_reuse/
Dmem_swap_manager.cc214 auto kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(user.first, IntToSize(user.second)); in IsInplaceRelevantOp() local
247 …auto kernel_with_index = common::AnfAlgo::GetPrevNodeOutput(user_kernel, IntToSize(node_pair.secon… in SaveUserKernelTopoOrder() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dacl_stream_assign.cc120 auto kernel_with_index = common::AnfAlgo::VisitKernelWithReturnType(input_node, 0, true); in AssignStream() local
245 auto kernel_with_index = common::AnfAlgo::VisitKernelWithReturnType(input_node, 0, true); in ProcessSideEffect() local

12