| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/gsm/ |
| D | mem_usage_analyzer.cc | 65 size_t tensor_id = 0; in AddTensorInfo() local
125 for (auto tensor_id : tensors) { in AddFusedTensorInfo() local
158 …auto add_tensor_usage = [this](size_t tensor_id, size_t kernel_id, size_t *kernel_mem, bool inplac… in AddKernelAndTensorInfo()
186 auto tensor_id = AddTensorInfo(prev_node_output.first, prev_node_output.second); in AddKernelAndTensorInfo() local
197 auto tensor_id = AddTensorInfo(real_node_pair.first, real_node_pair.second); in AddKernelAndTensorInfo() local
202 auto tensor_id = AddTensorInfo(node, index); in AddKernelAndTensorInfo() local
211 auto tensor_id = AddTensorInfo(node, index, true); in AddKernelAndTensorInfo() local
|
| D | swap_strategy_builder.cc | 262 void SwapStrategyBuilder::AddTensorAction(SwapActionType action_type, size_t tensor_id, size_t kern… in AddTensorAction()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/legacy_optimizer/graph/ |
| D | tensor_name_pass.cc | 35 auto tensor_id = node->inputIndex.at(i); in Run() local
44 auto tensor_id = node->outputIndex.at(i); in Run() local
|
| D | node_name_pass.cc | 72 auto tensor_id = node->inputIndex.at(i); in Run() local
84 auto tensor_id = node->outputIndex.at(i); in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/ |
| D | hook_py.cc | 43 const auto &tensor_id = GetTensorNumId(tensor.id()); in RegisterTensorBackwardHook() local
71 const auto &tensor_id = GetTensorNumId(id); in UpdateTensorBackwardHook() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/common/ |
| D | import_from_meta_graphT.cc | 29 AnfNodePtr AnfImporterFromMetaGraphT::GetNode(int tensor_id) { in GetNode()
37 void AnfImporterFromMetaGraphT::AddNode(int tensor_id, AnfNodePtr node) { nodes_[tensor_id] = std::… in AddNode()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/ |
| D | tensorrt_graph_executor.cc | 102 TensorInfo KernelTensorAsTensorInfo(const session::KernelWithIndex &tensor_id) { in KernelTensorAsTensorInfo()
151 for (auto &tensor_id : input_nodes) { in GetAbstractArgsFromCNode() local
166 session::KernelWithIndex tensor_id = {cnode, output_idx}; in GetAbstractArgsFromCNode() local
189 for (auto &tensor_id : inputs) { in GetModelInputsInfo() local
210 for (auto &tensor_id : outputs) { in GetModelOutputsInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/profiler/ |
| D | memory_profiling.proto | 44 uint64 tensor_id = 1; // tensor id field
|
| D | memory_profiling.h | 59 void SetTensorId(uint64_t tensor_id) { tensor_id_ = tensor_id; } in SetTensorId()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/ascend_native/ |
| D | ascend_native_composite_kernel.cc | 139 static inline kernel::InferTensor *anfTensorToTensorInfo(const common::KernelWithIndex &tensor_id) { in anfTensorToTensorInfo()
173 for (auto &tensor_id : input_nodes) { in CreateInputKernelTensors() local
230 mindspore::common::KernelWithIndex tensor_id = {cnode, output_idx}; in CreateOutputKernelTensors() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/ascend_native/ |
| D | delegate.cc | 48 static inline kernel::InferTensor *anfTensorToTensorInfo(const common::KernelWithIndex &tensor_id) { in anfTensorToTensorInfo()
197 for (auto &tensor_id : input_nodes) { in CreateInputKernelTensors() local
221 common::KernelWithIndex tensor_id = {cnode, output_idx}; in CreateOutputKernelTensors() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/ |
| D | hook.cc | 75 TensorBackwardHook::TensorBackwardHook(uint64_t tensor_id, const py::function &obj) { in TensorBackwardHook()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/common/draw/ |
| D | graphviz_graph_builder.cc | 84 auto tensor_id = node_id + "_in_" + std::to_string(i); in Build() local
105 auto tensor_id = GetTensorId(tensor); in AppendGraphInputNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/hardware/ |
| D | gpu_somas.cc | 122 auto tensor_id = tensor->GetId(); in InplaceNodeProcess() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/memory_offload/ |
| D | infer_strategy_builder.cc | 153 …yOffloadInferStrategyBuilder::AddTensorAction(device::SwapActionType action_type, size_t tensor_id, in AddTensorAction()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/somas/ |
| D | somas.h | 48 size_t tensor_id; member
|
| D | somas.cc | 495 const auto &tensor_id = tensor_json[kTensorId]; in VerifySomasResult() local
544 const auto &tensor_id = tensor_json[kTensorId]; in UpdateTensorsOffset() local
1477 for (auto tensor_id : contiguous_list) { in UpdateContiguousTensorList() local
2146 … bool is_peak = std::any_of(ref_node_list.begin(), ref_node_list.end(), [this](size_t tensor_id) { in CalcLowerBound()
2150 for (size_t tensor_id : ref_node_list) { in CalcLowerBound() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/ |
| D | mem_swap_scheduler.cc | 158 [&](size_t tensor_id) { return tensors_id_index_map.at(tensor_id); }); in GenActionIndexList()
161 const auto tensor_id = tensor_info->tensor_id_; in GenActionIndexList() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/profiler/parser/ |
| D | memory_usage_parser.py | 378 def _update_mem_change(self, size, start, end, tensor_id): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/ |
| D | func_graph_utils.cc | 178 session::KernelWithIndex tensor_id = {cnode, output_idx}; in GetCNodeInputsOutputs() local
|