| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/ |
| D | data_source_actor.cc | 116 std::vector<DeviceTensor *> device_tensors; in FillDataBuffer() local
126 auto &device_tensors = buffers_.back(); in SendMemoryAllocReq() local
138 auto &device_tensors = buffers_.front(); in SendMemoryFreeReq() local
164 auto &device_tensors = buffers_.back(); in OnMemoryAllocFinish() local
228 std::vector<DeviceTensor *> device_tensors; in FillDataBuffer() local
243 auto &device_tensors = buffers_.back(); in SendMemoryAllocReq() local
268 auto &device_tensors = buffers_.front(); in SendMemoryFreeReq() local
305 auto &device_tensors = buffers_.back(); in OnMemoryAllocFinish() local
|
| D | any_type_kernel_actor.cc | 221 std::string GenerateIDForGraph(const std::vector<DeviceTensor *> &device_tensors, const std::vector… in GenerateIDForGraph()
270 …rAbstractForModelGraph(const KernelGraphPtr &graph, const std::vector<DeviceTensor *> &device_tens… in InferParameterAbstractForModelGraph()
|
| D | data_prepare_actor.cc | 732 const auto &device_tensors = DeviceTensorStore::GetInstance().Fetch(front_node.get()); in PrepareDataForDeviceTensorStore() local
1187 const auto &device_tensors = DeviceTensorStore::GetInstance().Fetch(front_node.get()); in CopyDataFromDeviceTensorStore() local
1365 auto device_tensors = DeviceTensorStore::GetInstance().Fetch(front_parameter.get()); in PrepareDeviceTensorStoreForControlNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/memory/ |
| D | memory_swap_actor.cc | 63 std::vector<DeviceTensor *> device_tensors; in GetDeviceTensors() local
74 void MemorySwapActor::AllocDeviceContinuousMem(const std::vector<DeviceTensor *> &device_tensors) { in AllocDeviceContinuousMem()
106 const std::vector<DeviceTensor *> &device_tensors) { in Swap()
128 const auto &device_tensors = GetDeviceTensors(device_tensor_indexes); in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/control_flow/ |
| D | control_actor.cc | 53 …r::GetAllDeviceTensors(const OpPartialPtr &op_partial, std::vector<DeviceTensor *> *device_tensors… in GetAllDeviceTensors()
66 std::vector<DeviceTensor *> *device_tensors) { in GetAllDeviceTensors()
242 …auto device_tensors = DeviceTensorStore::GetInstance().Fetch(device_tensor_store_key.second.get()); in FetchInput() local
|
| D | entrance_actor.cc | 117 …const auto &device_tensors = real_parameters_with_branch_id_[sequential_num].front().device_tensor… in FetchInput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/data_dump/ |
| D | dump_utils.cc | 97 const auto &device_tensors = DeviceTensorStore::GetInstance().Fetch(node.get()); in GetParameterInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/debugger/ |
| D | debugger_utils.cc | 99 void LoadInputs(const CNodePtr &cnode, std::vector<device::DeviceAddress *> device_tensors, uint32_… in LoadInputs()
155 void LoadOutputs(const CNodePtr &cnode, std::vector<device::DeviceAddress *> device_tensors, uint32… in LoadOutputs()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/ |
| D | memory_swap_node_scheduler.cc | 220 std::vector<DeviceTensor *> device_tensors; in CollectKernelInputMemStatistic() local
258 std::vector<DeviceTensor *> device_tensors; in CollectKernelOutputMemStatistic() local
437 std::vector<DeviceTensor *> device_tensors(continuous_info->align_size_list_.size(), nullptr); in GenSwapInActor() local
|
| D | embedding_cache_scheduler.cc | 415 … std::vector<DeviceTensorPtr> device_tensors = DeviceTensorStore::GetInstance().Fetch(param.get()); in InitEmbeddingStorage() local
|
| D | scheduler_helper.cc | 1047 auto &device_tensors = ref_node_formal_parameter_device_tensor.second; in CheckControlActorValid() local
1059 auto &device_tensors = ref_formal_parameter_device_tensor.second; in CheckControlActorValid() local
|
| D | mem_swap_scheduler.cc | 254 const auto &device_tensors = DeviceTensorStore::GetInstance().Fetch(parameter.get()); in AddSwappableRootParameter() local
|
| D | any_type_graph_scheduler.cc | 346 auto device_tensors = DeviceTensorStore::GetInstance().Fetch(node.get()); in PrepareDataForValueNode() local
|
| D | graph_scheduler.cc | 2901 …auto device_tensors = DeviceTensorStore::GetInstance().Fetch(auto_monad_device_tensor_store.get()); in LinkDeviceTensorStoreForAutoMonadActor() local
3240 const auto device_tensors = DeviceTensorStore::GetInstance().Fetch(front_node.get()); in DumpDeviceTensorStore() local
3266 const auto device_tensors = DeviceTensorStore::GetInstance().Fetch(front_node.get()); in DumpDeviceTensorStore() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/data_queue/ |
| D | data_queue_mgr.cc | 255 std::vector<std::shared_ptr<device::DeviceAddress>> device_tensors; in UpdateGetNextWithDataQueueItems() local
|