Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_simple_mem_plan.cc30 auto kernel_with_index = AnfAlgo::GetPrevNodeOutput(kernel, i); in MemPlan() local
74 auto kernel_with_index = AnfAlgo::GetPrevNodeOutput(kernel, i); in MemAssign() local
Dcpu_kernel_runtime.cc248 const session::KernelWithIndex &kernel_with_index, in CreatTensorForOutput()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/
Dtbe_json_utils.cc81 session::KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(anf_node, real_idx); in GetInputOriShapeForTbeBuild() local
88 session::KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(anf_node, real_idx); in GetInputDeviceShapeForTbeBuild() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dinsert_tensor_move_for_hccl_op.cc34 auto kernel_with_index = AnfAlgo::VisitKernelWithReturnType(node, 0, true); in IsParameterOrValueNode() local
79 auto kernel_with_index = AnfAlgo::VisitKernelWithReturnType(input, 0, false); in NeedInsertTensorMove() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dtsa_atomic_add_to_first_tensor.cc104 auto kernel_with_index = AnfAlgo::VisitKernel(tsa_first_input, 0); in ProcessTsaFirstNode() local
168 auto kernel_with_index = AnfAlgo::VisitKernel(modified_input, 0); in CorrectKernelBuildInfo() local
Dgraph_kernel_helper.cc325 auto kernel_with_index = AnfAlgo::VisitKernel(inputs[i], 0); in SetNewKernelInfo() local
355 auto kernel_with_index = AnfAlgo::VisitKernel(new_outputs[i], 0); in SetNewKernelInfo() local
393 auto kernel_with_index = AnfAlgo::VisitKernel(inputs[i], 0); in CreateNewFuseCNode() local
727 auto kernel_with_index = AnfAlgo::VisitKernel(inputs[i], 0); in CreateCNode() local
Dadd_atomic_clean.cc255 auto kernel_with_index = AnfAlgo::VisitKernel(new_input, 0); in CorrectKernelBuildInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Ddeal_ref_and_split_unsupported_transdata.cc30 session::KernelWithIndex kernel_with_index = AnfAlgo::VisitKernel(node, 0); in FindRefOriginNode() local
71 …session::KernelWithIndex kernel_with_index = AnfAlgo::VisitKernel(AnfAlgo::GetInputNode(cnode, inp… in AddRefNodePairToKernelGraph() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.cc797 KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(anf_node, input_idx); in GetPrevNodeOutputFormat() local
802 KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(node, input_idx); in GetPrevNodeOutputReshapeType() local
851 KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(node, input_idx); in GetPrevNodeOutputInferShape() local
973 KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(node, input_idx); in GetPrevNodeOutputInferDataType() local
1024 KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(anf_node, input_idx); in GetPrevNodeOutputDeviceDataType() local
1084 auto kernel_with_index = AnfAlgo::GetPrevNodeOutput(cnode, 0); in OutputAddrExist() local
1105 KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(anf_node, input_idx); in GetPrevNodeOutputAddr() local
1111 KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(anf_node, input_idx); in GetPrevNodeMutableOutputAddr() local
1201 KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(node, input_idx); in GetPrevNodeOutputDetailShape() local
1455 AnfNodePtr AnfRuntimeAlgorithm::GetOutputOfGraphkernel(const KernelWithIndex &kernel_with_index) { in GetOutputOfGraphkernel()
[all …]
Dsession_basic.cc1293 auto kernel_with_index = AnfAlgo::VisitKernel(input, 0); in GetRefCount() local
1307 for (auto &kernel_with_index : input_kernel) { in HandleOpInputs() local
1346 auto kernel_with_index = make_pair(kernel, out_index++); in HandleOpOutputs() local
1423 TensorPtr SessionBasic::GetCNodeOutputTensor(const KernelWithIndex &kernel_with_index, in GetCNodeOutputTensor()
1442 auto kernel_with_index = AnfAlgo::VisitKernel(input, 0); in GetOpInputTensors() local
1480 auto kernel_with_index = AnfAlgo::VisitKernel(input, 0); in GetOpInputTensorByIndex() local
Dascend_session.cc166 TensorPtr GetCNodeOutputStubTensor(const KernelWithIndex &kernel_with_index, in GetCNodeOutputStubTensor()
187 KernelWithIndex kernel_with_index = std::make_pair(kernel, out_idx++); in GenOpOutputStubTensor() local
1089 auto kernel_with_index = AnfAlgo::VisitKernel(input, 0); in GetOpInputStubTensors() local
/third_party/mindspore/mindspore/ccsrc/debug/data_dump/
Dcpu_e2e_dump.cc63 auto kernel_with_index = AnfAlgo::GetPrevNodeOutput(node, j); in DumpInputImpl() local
De2e_dump.cc169 auto kernel_with_index = AnfAlgo::GetPrevNodeOutput(node, j); in DumpInputImpl() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_swap_manager.cc214 auto kernel_with_index = AnfAlgo::GetPrevNodeOutput(user.first, IntToSize(user.second)); in IsInplaceRelevantOp() local
247 … auto kernel_with_index = AnfAlgo::GetPrevNodeOutput(user_kernel, IntToSize(node_pair.second - 1)); in SaveUserKernelTopoOrder() local
Dmem_reuse_checker.cc384 auto kernel_with_index = AnfAlgo::VisitKernel(input, 0); in CheckNormalIR() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc704 …session::KernelWithIndex kernel_with_index = AnfAlgo::VisitKernel(AnfAlgo::GetInputNode(assign_add… in CreateStreamAssignAddnOP() local
987 …session::KernelWithIndex kernel_with_index = AnfAlgo::VisitKernel(AnfAlgo::GetInputNode(assign_add… in CreateAssignAdd() local
1036 …session::KernelWithIndex kernel_with_index = AnfAlgo::VisitKernel(AnfAlgo::GetInputNode(assign_cno… in CreateAssign() local
Dkernel_runtime.cc283 KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(kernel, input_index, true); in ResetNodeAddress() local
638 auto kernel_with_index = AnfAlgo::VisitKernelWithReturnType(node, 0, true); in AssignStaticMemoryOutput() local
1315 auto kernel_with_index = AnfAlgo::GetPrevNodeOutput(kernel, real_input, true); in AssignKernelAddress() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_select_graph_kernel.cc66 auto kernel_with_index = AnfAlgo::VisitKernel(input_kernel_node, 0); in ResetKernelBuildInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_helper.cc430 auto kernel_with_index = AnfAlgo::VisitKernelWithReturnType(cur_input, 0); in InsertCastForInput() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_scheduler.cc896 const auto &kernel_with_index = GetFrontNodeByKernelGraph(from_kernel, graph); in LinkDataArrow() local
1262 auto kernel_with_index = AnfAlgo::GetPrevNodeOutput(skipped_node, i, false); in LinkControlArrowBySkippedNode() local
1858 const auto &kernel_with_index = AnfAlgo::VisitKernelWithReturnType(input_node, 0); in LinkDataArrowByControlNode() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_build.cc167 session::KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(anf_node, real_index); in GetInputShapeForTbeBuild() local
175 session::KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(anf_node, real_index); in GetInputDeviceShapeForTbeBuild() local
/third_party/mindspore/mindspore/ccsrc/vm/
Dbackend.cc524 auto kernel_with_index = AnfAlgo::VisitKernel(input_node, 0); in GetControlOpInput() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc441 …auto kernel_with_index = AnfAlgo::GetPrevNodeOutput(output_info.first, IntToSize(output_info.secon… in GetRealNodeUsedListByOutputIdx() local
/third_party/mindspore/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc41 KernelWithIndex kernel_with_index; in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.cc1408 auto kernel_with_index = AnfAlgo::GetPrevNodeOutput(node, i); in GetPrevNodeOutput() local