| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_compiler.cc | 99 auto device_address = device_context->CreateDeviceAddress(nullptr, tensor_size, in CreateParameterDeviceAddress()
140 device_context->CreateDeviceAddress(nullptr, tensor_size, output_format, output_type_id); in CreateDeviceAddressForTensorValue()
162 …auto address = device_context->CreateDeviceAddress(nullptr, tensor_size, kOpFormat_DEFAULT, kNumbe… in CreateValueNodeDeviceAddress()
189 …auto device_address = device_context->CreateDeviceAddress(nullptr, output_sizes[i], output_format,… in CreateKernelOutputDeviceAddress()
208 …auto device_address = device_context->CreateDeviceAddress(nullptr, workspace_sizes[i], "", kTypeUn… in CreateKernelWorkspaceDeviceAddress()
345 CreateDeviceAddress(graph, device_context); in CompileGraphImpl()
411 CreateDeviceAddress(graph, device_context); in CompileGraph()
442 void GraphCompiler::CreateDeviceAddress(const KernelGraphPtr &graph, const DeviceContext *device_co… in CreateDeviceAddress() function in mindspore::runtime::GraphCompiler
|
| D | graph_compiler.h | 178 void CreateDeviceAddress(const KernelGraphPtr &graph, const DeviceContext *device_context) const;
|
| D | control_node_parser.cc | 169 device_context->CreateDeviceAddress(nullptr, tensor_size, output_format, output_type_id); in CreateDeviceTensorForValueNode()
193 …device::DeviceAddressPtr address = device_context->CreateDeviceAddress(nullptr, size, kOpFormat_DE… in CreateDeviceTensorForFrontParameter()
|
| D | graph_scheduler.cc | 1160 copy_actor->output_ = to_device_context->CreateDeviceAddress( in LinkDataArrowForCopyActor()
2276 auto other_type_device_tensor = device_context->CreateDeviceAddress( in PersistDeviceTensor()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_kernel_runtime.h | 54 DeviceAddressPtr CreateDeviceAddress(void *device_ptr, size_t device_size, const string &format,
56 …DeviceAddressPtr CreateDeviceAddress(void *device_ptr, size_t device_size, const string &format, T…
|
| D | cpu_kernel_runtime.cc | 114 address = CreateDeviceAddress(nullptr, tensor_size, kOpFormat_DEFAULT, output_type_id); in AssignValueNodeAddress()
148 auto address = CreateDeviceAddress(nullptr, tensor_size, format, output_type_id); in AssignInputNodeAddress()
165 …AnfAlgo::SetOutputAddr(CreateDeviceAddress(nullptr, output_sizes[i], output_format, output_type), … in AssignKernelOutputAddress()
170 …AnfAlgo::SetWorkspaceAddr(CreateDeviceAddress(nullptr, workspace_sizes[i], kOpFormat_DEFAULT, kNum… in AssignKernelOutputAddress()
176 DeviceAddressPtr CPUKernelRuntime::CreateDeviceAddress(void *device_ptr, size_t device_size, const … in CreateDeviceAddress() function in mindspore::device::cpu::CPUKernelRuntime
181 DeviceAddressPtr CPUKernelRuntime::CreateDeviceAddress(void *device_ptr, size_t device_size, const … in CreateDeviceAddress() function in mindspore::device::cpu::CPUKernelRuntime
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 188 …CreateDeviceAddress(nullptr, runtime_info->output_tensor_size(i), output_format, output_type, {nod… in RunOpGetCommunicationOutputInfo()
235 auto device_address = CreateDeviceAddress(nullptr, tensor_size, output_format, output_type); in RunOpMallocPre()
267 … CreateDeviceAddress(nullptr, output_tensor_size, output_format, output_type_id, {item, index}); in RunOpMallocPre()
295 …auto device_address = CreateDeviceAddress(nullptr, tensor_size, AnfAlgo::GetOutputFormat(input_nod… in ResetNodeAddress()
304 …AnfAlgo::SetOutputAddr(CreateDeviceAddress(nullptr, output_sizes[i], output_format, output_type), … in ResetNodeAddress()
309 …AnfAlgo::SetWorkspaceAddr(CreateDeviceAddress(nullptr, workspace_sizes[i], kOpFormat_DEFAULT, kNum… in ResetNodeAddress()
424 …CreateDeviceAddress(nullptr, tensor_size, AnfAlgo::GetOutputFormat(item, index), output_type_id, {… in RunOpAssignInputMemory()
473 …auto device_address = CreateDeviceAddress(nullptr, output_sizes[i], output_format, output_type, {k… in RunOpAssignOutputMemory()
492 auto device_address = CreateDeviceAddress(nullptr, workspace_lists[i], "", kTypeUnknown); in RunOpAssignWorkSpaceMemory()
612 …device_address = CreateDeviceAddress(address.addr, address.size, AnfAlgo::GetOutputFormat(item, in… in AssignStaticMemoryInput()
[all …]
|
| D | kernel_runtime.h | 122 …virtual DeviceAddressPtr CreateDeviceAddress(void *device_ptr, size_t device_size, const string &f…
124 …virtual DeviceAddressPtr CreateDeviceAddress(void *device_ptr, size_t device_size, const string &f…
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_kernel_runtime.h | 76 DeviceAddressPtr CreateDeviceAddress(void *device_ptr, size_t device_size, const string &format,
78 …DeviceAddressPtr CreateDeviceAddress(void *device_ptr, size_t device_size, const string &format, T…
|
| D | ascend_kernel_runtime.cc | 403 DeviceAddressPtr AscendKernelRuntime::CreateDeviceAddress(void *device_ptr, size_t device_size, con… in CreateDeviceAddress() function in mindspore::device::ascend::AscendKernelRuntime
408 DeviceAddressPtr AscendKernelRuntime::CreateDeviceAddress(void *device_ptr, size_t device_size, con… in CreateDeviceAddress() function in mindspore::device::ascend::AscendKernelRuntime
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.h | 57 DeviceAddressPtr CreateDeviceAddress(void *device_ptr, size_t device_size, const string &format,
59 …DeviceAddressPtr CreateDeviceAddress(void *device_ptr, size_t device_size, const string &format, T…
|
| D | gpu_kernel_runtime.cc | 227 DeviceAddressPtr GPUKernelRuntime::CreateDeviceAddress(void *device_ptr, size_t device_size, const … in CreateDeviceAddress() function in mindspore::device::gpu::GPUKernelRuntime
232 DeviceAddressPtr GPUKernelRuntime::CreateDeviceAddress(void *device_ptr, size_t device_size, const … in CreateDeviceAddress() function in mindspore::device::gpu::GPUKernelRuntime
628 … auto device_address = CreateDeviceAddress(nullptr, output_sizes[i], output_format, output_type); in InitKernelOutputAddress()
642 auto device_address = CreateDeviceAddress(nullptr, workspace_sizes[i], "", kTypeUnknown); in InitKernelWorkspaceAddress()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/ |
| D | cpu_device_context.h | 43 …DeviceAddressPtr CreateDeviceAddress(void *const device_ptr, size_t device_size, const string &for…
|
| D | cpu_device_context.cc | 102 DeviceAddressPtr CPUDeviceContext::CreateDeviceAddress(void *const device_ptr, size_t device_size, … in CreateDeviceAddress() function in mindspore::device::cpu::CPUDeviceContext
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/ |
| D | device_context.h | 71 …virtual DeviceAddressPtr CreateDeviceAddress(void *const device_ptr, size_t device_size, const str…
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/ |
| D | gpu_device_context.h | 47 …DeviceAddressPtr CreateDeviceAddress(void *const device_ptr, size_t device_size, const string &for…
|
| D | gpu_device_context.cc | 199 DeviceAddressPtr GPUDeviceContext::CreateDeviceAddress(void *const device_ptr, size_t device_size, … in CreateDeviceAddress() function in mindspore::device::gpu::GPUDeviceContext
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | output_actor.cc | 95 … auto new_device_tensor = device_context->CreateDeviceAddress(nullptr, device_tensor->GetSize(), in UpdateOutputDeviceAddress()
|
| D | data_prepare_actor.cc | 314 auto device_address = device_context->CreateDeviceAddress( in PrepareDataForStepMode()
418 host_tensor_address = device_context->CreateDeviceAddress(nullptr, device_tensor->GetSize(), in PrepareDataForWeightNode()
|
| D | kernel_actor.cc | 278 copy_input_device_tensors_[input_data->index_] = device_contexts_[0]->CreateDeviceAddress( in CopyInputDeviceTensor()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | backend.cc | 252 … auto new_device_tensor = device_context->CreateDeviceAddress(nullptr, device_tensor->GetSize(), in UpdateOutputDeviceAddress()
|