| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | data_prepare_actor.cc | 31 void SyncTensorData(const TensorPtr &host_tensor, const DeviceTensorPtr &device_tensor, const AnfNo… in SyncTensorData()
75 const auto &device_tensor = AnfAlgo::GetPrevNodeMutableOutputAddr(node, i, false); in FetchContinuousMemoryInfo() local
84 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(node, i, false); in FetchContinuousMemoryInfo() local
318 auto device_tensor = AnfAlgo::GetMutableOutputAddr(input_node, 0, false); in PrepareDataForStepMode() local
349 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(node, i, false); in PrepareDataForValueNodeTensor() local
376 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(node, 0, false); in PrepareDataForValueNode() local
412 auto device_tensor = AnfAlgo::GetMutableOutputAddr(backend_node, 0, false); in PrepareDataForWeightNode() local
480 for (auto &device_tensor : device_tensors) { in PrepareDataForControlWeightNode() local
500 for (const auto &device_tensor : device_tensors) { in PrepareDataForControlWeightNode() local
|
| D | kernel_actor.cc | 157 for (auto &device_tensor : alloc_list) { in AllocateMemory() local
172 for (auto &device_tensor : free_list) { in FreeMemory() local
257 …const auto &device_tensor = std::dynamic_pointer_cast<DeviceTensor>(input_tensor->device_address()… in PushInputDeviceTensor() local
324 … auto device_tensor = DeviceTensorStore::GetInstance().Fetch(device_tensor_store_key.second.get(), in FetchInputDeviceTensor() local
|
| D | output_actor.cc | 38 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(output_node, output_index, false); in CreateOutputTensor() local
88 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(output_node, output_index, false); in UpdateOutputDeviceAddress() local
|
| D | memory_manager_actor.cc | 94 auto &device_tensor = (*alloc_list)[i]; in AllocateBatchMemory() local
146 auto &device_tensor = (*free_list)[i]; in FreeBatchMemory() local
|
| D | actor_common.cc | 170 void UpdateRefCount(DeviceTensor *const device_tensor, bool is_max_ref_count) { in UpdateRefCount()
182 auto device_tensor = AnfAlgo::GetMutableOutputAddr(node, output_idx, false); in UpdateRefCount() local
|
| D | data_source_actor.cc | 129 for (auto &device_tensor : kernel_info_->output_address_list()) { in FillDataBuffer() local
274 auto &device_tensor = device_tensors[i]; in OnMemoryAllocFinish() local
|
| D | switch_actor.cc | 280 DeviceTensor *device_tensor = input_device_tensors_[0]; in GetIndex() local
361 auto device_tensor = in FetchInputDeviceTensor() local
|
| D | gather_actor.cc | 154 auto device_tensor = in FetchInputDeviceTensor() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | device_tensor_store.h | 84 for (const auto &device_tensor : iter->second) { in Fetch() local
|
| D | graph_scheduler.cc | 1343 auto device_tensor = AnfAlgo::GetPrevNodeMutableOutputAddr(from_allreduce_node, i, false); in LinkControlArrowBySendRecvNodes() local
1873 …auto device_tensor = AnfAlgo::GetMutableOutputAddr(from_actor->kernel_, backend_node.second, false… in LinkDataArrowByControlNode() local
1881 …auto device_tensor = AnfAlgo::GetMutableOutputAddr(from_actor->kernel_, input_with_index.second, f… in LinkDataArrowByControlNode() local
1907 …auto device_tensor = AnfAlgo::GetMutableOutputAddr(from_actor->data_nodes_[iter->second], 0, false… in LinkDataArrowByControlNode() local
2235 auto device_tensor = AnfAlgo::GetMutableOutputAddr(value_node, 0, false); in PersistDeviceTensor() local
2261 auto device_tensor = AnfAlgo::GetMutableOutputAddr(input_node, 0, false); in PersistDeviceTensor() local
2288 auto device_tensor = AnfAlgo::GetMutableOutputAddr(value_node.first, 0, false); in PersistDeviceTensor() local
2523 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(data_kernel, i, false); in DumpDSActor() local
2536 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(data_node, 0, false); in DumpDSActor() local
2568 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(kernel, i, false); in DumpKernelActor() local
[all …]
|
| /third_party/mindspore/tests/st/cpp/model/ |
| D | test_zero_copy.cc | 140 MSTensor *device_tensor = in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | backend.cc | 209 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(output_node, output_index, false); in CreateOutputTensor() local
245 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(output_node, output_index, false); in UpdateOutputDeviceAddress() local
|