| /third_party/mindspore/mindspore/core/ir/ |
| D | device_sync.h | 36 virtual bool SyncHostToDevice(size_t size, const void *host_ptr) const { return true; } in SyncHostToDevice() function
40 …virtual bool SyncHostToDevice(const ShapeVector &shape, size_t size, TypeId type, const void *host…
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_device_address.cc | 67 bool GPUDeviceAddress::SyncHostToDevice(size_t size, const void *host_ptr) const { in SyncHostToDevice() function in mindspore::device::gpu::GPUDeviceAddress
98 bool GPUDeviceAddress::SyncHostToDevice(const ShapeVector &, size_t size, TypeId, const void *host_… in SyncHostToDevice() function in mindspore::device::gpu::GPUDeviceAddress
104 return SyncHostToDevice(size, host_ptr); in SyncHostToDevice()
|
| D | gpu_device_address.h | 42 bool SyncHostToDevice(size_t size, const void *host_ptr) const override;
44 bool SyncHostToDevice(const ShapeVector &shape, size_t size, TypeId type, const void *host_ptr,
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_device_address.h | 46 bool SyncHostToDevice(size_t size, const void *host_ptr) const override;
48 bool SyncHostToDevice(const ShapeVector &shape, size_t size, TypeId type, const void *host_ptr,
|
| D | ascend_device_address.cc | 188 bool AscendDeviceAddress::SyncHostToDevice(size_t size, const void *host_ptr) const { in SyncHostToDevice() function in mindspore::device::ascend::AscendDeviceAddress
360 bool AscendDeviceAddress::SyncHostToDevice(const ShapeVector &shape, size_t size, mindspore::TypeId… in SyncHostToDevice() function in mindspore::device::ascend::AscendDeviceAddress
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_device_address.h | 40 bool SyncHostToDevice(const ShapeVector &shape, size_t size, TypeId type, const void *host_ptr,
|
| D | cpu_device_address.cc | 101 bool CPUDeviceAddress::SyncHostToDevice(const ShapeVector &, size_t size, TypeId type, const void *… in SyncHostToDevice() function in mindspore::device::cpu::CPUDeviceAddress
|
| D | cpu_kernel_runtime.cc | 120 …if (!address->SyncHostToDevice(data_shape, LongToSize(tensor->data().nbytes()), tensor->data_type(… in AssignValueNodeAddress()
342 …if (!address->SyncHostToDevice(data_shape, LongToSize(tensor->data().nbytes()), tensor->data_type(… in BindInputTensorAddressPtr()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | dynamic_shape_kernel.cc | 46 output_addr->SyncHostToDevice(output_shape, LongToSize(output_tensor_for_sync->data().nbytes()), in Execute()
|
| D | dynamic_broadcast_gradient_args_kernel.cc | 181 …if (!out_addr->SyncHostToDevice(out_shape, LongToSize(tensor_for_sync->data().nbytes()), tensor_fo… in SetOutputValue()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | ascend_inference_session.cc | 48 if (!device_address->SyncHostToDevice(trans::GetRuntimePaddingShape(pk_node, 0), in LoadInputData()
77 if (!device_address->SyncHostToDevice(trans::GetRuntimePaddingShape(pk_node, 0), in CompileGraphImpl()
|
| D | gpu_inference_session.cc | 48 if (!device_address->SyncHostToDevice(trans::GetRuntimePaddingShape(pk_node, 0), in LoadInputData()
77 if (!device_address->SyncHostToDevice(trans::GetRuntimePaddingShape(pk_node, 0), in CompileGraphImpl()
|
| D | ascend_session.cc | 244 …if (!device_address->SyncHostToDevice(cur_loop_tensor->shape(), LongToSize(cur_loop_tensor->data()… in LoadCtrlInputTensor()
263 …if (!device_address->SyncHostToDevice(next_loop_tensor->shape(), LongToSize(next_loop_tensor->data… in LoadCtrlInputTensor()
280 …if (!device_address->SyncHostToDevice(epoch_tensor->shape(), LongToSize(epoch_tensor->data().nbyte… in LoadCtrlInputTensor()
499 …!device_address->SyncHostToDevice(trans::GetRuntimePaddingShape(input_node, 0), size, tensor->data… in LoadInputData()
1748 if (!addr->SyncHostToDevice(trans::GetRuntimePaddingShape(backend_parameter, 0), tensor_size, in SyncInitialTenosrToDevice()
|
| D | gpu_session.cc | 365 …if (!device_address->SyncHostToDevice(trans::GetRuntimePaddingShape(pk_node, 0), size, tensor->dat… in LoadInputData()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | actor_common.cc | 159 return dst_device_tensor->SyncHostToDevice(copy_size, src_device_tensor->GetPtr()); in Copy()
|
| D | data_prepare_actor.cc | 49 …if (!device_tensor->SyncHostToDevice(trans::GetRuntimePaddingShape(node, 0), host_tensor_size, hos… in SyncTensorData()
393 if (!device_tensor->SyncHostToDevice(shape, tensor_size, kNumberTypeUInt8, value.data())) { in PrepareDataForValueNode()
|
| D | data_source_actor.cc | 287 if (!device_tensor->SyncHostToDevice(trans::GetRuntimePaddingShape(data_nodes_[i], 0), in OnMemoryAllocFinish()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_adjust.cc | 755 if (!device_address->SyncHostToDevice(trans::GetRuntimePaddingShape(pk_node, 0), in StepLoadCtrlInputs()
842 …if (!device_address->SyncHostToDevice(tensor->shape(), LongToSize(tensor->data().nbytes()), tensor… in InitCtrlInputs()
|
| D | kernel_runtime.cc | 992 …if (!address->SyncHostToDevice(trans::GetRuntimePaddingShape(value_node, 0), tensor_size, tensor->… in AssignValueNodeTensor()
1056 if (!address->SyncHostToDevice(shape, tensor_size, kNumberTypeUInt8, value.data())) { in AssignStaticMemoryValueNode()
|