Searched refs:input_kernel (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/debug/debugger/ |
| D | debugger_utils.cc | 60 auto input_kernel = cnode->input(j + 1); in LoadInputs() local
61 std::string input_kernel_name = GetKernelNodeName(input_kernel); in LoadInputs()
63 auto type = AnfAlgo::GetOutputInferDataType(input_kernel, PARAMETER_OUTPUT_INDEX); in LoadInputs()
72 ShapeVector int_shapes = trans::GetRuntimePaddingShape(input_kernel, PARAMETER_OUTPUT_INDEX); in LoadInputs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/ |
| D | kernel_refcount.h | 87 …nsertInputKernel(const std::shared_ptr<KernelDef> &input_kernel) { input_kernels_.insert(input_ker… in InsertInputKernel() argument
|
| /third_party/mindspore/mindspore/ccsrc/debug/data_dump/ |
| D | e2e_dump.cc | 181 auto input_kernel = node->input(j + 1); in DumpInputImpl() local
182 std::string input_kernel_name = GetKernelNodeName(input_kernel); in DumpInputImpl()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 163 auto input_kernel = kernel->input(j + 1); in LoadKernelData() local
164 MS_EXCEPTION_IF_NULL(input_kernel); in LoadKernelData()
165 std::string input_kernel_name = GetKernelNodeName(input_kernel); in LoadKernelData()
167 auto type = AnfAlgo::GetOutputInferDataType(input_kernel, PARAMETER_OUTPUT_INDEX); in LoadKernelData()
176 ShapeVector int_shapes = trans::GetRuntimePaddingShape(input_kernel, PARAMETER_OUTPUT_INDEX); in LoadKernelData()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.h | 77 std::set<KernelWithIndex> input_kernel; member
192 …void HandleOpInputs(const std::set<KernelWithIndex> &input_kernel, std::map<KernelWithIndex, size_…
|
| D | session_basic.cc | 1302 void SessionBasic::HandleOpInputs(const std::set<KernelWithIndex> &input_kernel, in HandleOpInputs() argument
1307 for (auto &kernel_with_index : input_kernel) { in HandleOpInputs()
1455 input_tensor_info->input_kernel.insert(kernel_with_index); in GetOpInputTensors()
1491 input_tensor_info->input_kernel.insert(kernel_with_index); in GetOpInputTensorByIndex()
2307 HandleOpInputs(input_tensor_info.input_kernel, &cnode_refcount, &op_output_map); in RunOpsInGraphImpl()
|
| /third_party/mindspore/mindspore/ccsrc/debug/ |
| D | debug_services.cc | 1237 auto input_kernel = kernel->input(j + 1); in IsWatchPointNodeInput() local
1238 std::string input_kernel_name = GetKernelNodeName(input_kernel); in IsWatchPointNodeInput()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | backend.cc | 761 … graph_compiler_->UpdateRefCount(input_tensor_info.input_kernel, &cnode_ref_count, &op_output_map); in RunGraphBySingleOp()
|