Searched refs:tensor_address (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | data_prepare_actor.cc | 254 auto tensor_address = std::dynamic_pointer_cast<DeviceTensor>(input_tensor->device_address()); in PrepareDataForHostTensorQueue() local
257 …if ((tensor_address != nullptr) && (tensor_address->DeviceType() == device_address->DeviceType()))… in PrepareDataForHostTensorQueue()
258 AnfAlgo::SetOutputAddr(tensor_address, 0, input_node.get()); in PrepareDataForHostTensorQueue()
567 auto tensor_address = std::dynamic_pointer_cast<DeviceTensor>(input_tensor->device_address()); in PrepareHostTensorQueueForControlNode() local
570 …if ((tensor_address != nullptr) && (tensor_address->DeviceType() == device_address->DeviceType()))… in PrepareHostTensorQueueForControlNode()
571 AnfAlgo::SetOutputAddr(tensor_address, 0, backend_node.get()); in PrepareHostTensorQueueForControlNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | gpu_session.cc | 344 … auto tensor_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in LoadInputData() local
347 if (tensor_address == nullptr || tensor_address != device_address) { in LoadInputData()
350 } else if (tensor->NeedSyncHostToDevice() || tensor_address == nullptr) { in LoadInputData()
352 } else if (tensor_address != device_address) { in LoadInputData()
353 if (tensor_address->DeviceType() == device_address->DeviceType()) { in LoadInputData()
354 AnfAlgo::SetOutputAddr(tensor_address, 0, pk_node.get()); in LoadInputData()
|
| D | cpu_session.cc | 182 auto tensor_address = tensor->device_address(); in LoadInputData() local
185 if (tensor_address == nullptr || tensor_address == address) { in LoadInputData()
192 if (std::dynamic_pointer_cast<device::DeviceAddress>(tensor_address)->DeviceType() != in LoadInputData()
|
| D | ascend_session.cc | 330 auto tensor_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in TensorNeedSync() local
331 if (tensor_address != device_address) { in TensorNeedSync()
333 NeedMemcpyInDevice(tensor_address, device_address)) { in TensorNeedSync()
335 … tensor_address->GetSize(), tensor_address->type_id(), in TensorNeedSync()
336 … tensor_address->GetPtr(), tensor_address->format()); in TensorNeedSync()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_kernel_runtime.cc | 326 auto tensor_address = tensor->device_address(); in BindInputTensorAddressPtr() local
327 if (AnfAlgo::IsParameterWeight(item->cast<ParameterPtr>()) && tensor_address != nullptr && in BindInputTensorAddressPtr()
328 tensor_address != address) { in BindInputTensorAddressPtr()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 936 auto tensor_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in AssignExtraStaticMem() local
938 << "Assign Static Memory for Output node, size:" << tensor_address->size(); in AssignExtraStaticMem()
939 …auto device_address = CreateDeviceAddress(nullptr, tensor_address->size(), tensor_address->format(… in AssignExtraStaticMem()
940 tensor_address->type_id(), {node, index}); in AssignExtraStaticMem()
942 …uint8_t *ptr = mem_manager_->MallocOutputMem(node, index, kStaticMem, tensor_address->size(), devi… in AssignExtraStaticMem()
1402 auto tensor_address = tensor->device_address(); in InitGraphInputTensors() local
1403 if (tensor_address != nullptr && tensor_address != device_address) { in InitGraphInputTensors()
|