Searched refs:output_address (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | kernel_actor.cc | 49 for (auto &output_address : kernel_info_->output_address_list()) { in Init() local
50 MS_EXCEPTION_IF_NULL(output_address); in Init()
51 (void)output_device_tensors_.emplace_back(output_address.get()); in Init()
52 (void)memory_alloc_list_.emplace_back(output_address.get()); in Init()
53 (void)memory_free_list_.emplace_back(output_address.get()); in Init()
355 auto output_address = output_addresses[i].get(); in FetchOutputDeviceTensor() local
356 MS_EXCEPTION_IF_NULL(output_address); in FetchOutputDeviceTensor()
357 if (output_size_list[i] != output_address->GetSize()) { in FetchOutputDeviceTensor()
359 output_address->SetSize(output_size_list[i]); in FetchOutputDeviceTensor()
363 if (output_device_tensors_[i] != output_address) { in FetchOutputDeviceTensor()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_info.cc | 48 bool KernelInfo::SetOutputAddr(const DeviceAddressPtr &output_address, size_t index) { in SetOutputAddr() argument
64 output_address_list_[index] = output_address; in SetOutputAddr()
91 bool KernelInfo::SetWorkspaceAddr(const DeviceAddressPtr &output_address, size_t index) { in SetWorkspaceAddr() argument
107 workspace_address_list_[index] = output_address; in SetWorkspaceAddr()
|
| D | kernel_info.h | 55 bool SetOutputAddr(const DeviceAddressPtr &output_address, size_t index);
59 bool SetWorkspaceAddr(const DeviceAddressPtr &output_address, size_t index);
|
| D | kernel_runtime.cc | 256 …auto output_address = std::dynamic_pointer_cast<device::DeviceAddress>(current_tensor->device_addr… in RunOpMallocPre() local
257 … if (output_address != nullptr && output_address->DeviceType() == GetTargetDeviceAddressType()) { in RunOpMallocPre()
258 AnfAlgo::SetOutputAddr(output_address, index, item.get()); in RunOpMallocPre()
407 …auto output_address = std::dynamic_pointer_cast<device::DeviceAddress>(current_tensor->device_addr… in RunOpAssignInputMemory() local
408 … if (output_address != nullptr && output_address->DeviceType() == GetTargetDeviceAddressType()) { in RunOpAssignInputMemory()
409 if (output_address->ptr_ == nullptr) { in RunOpAssignInputMemory()
410 if (!mem_manager_->MallocMemFromMemPool(output_address, output_address->size())) { in RunOpAssignInputMemory()
411 MS_LOG(EXCEPTION) << "Allocate memory failed, size:" << output_address->size(); in RunOpAssignInputMemory()
415 AnfAlgo::SetOutputAddr(output_address, index, item.get()); in RunOpAssignInputMemory()
965 … auto output_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in AssignValueNodeTensor() local
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/debug/ |
| D | print_gpu_kernel.h | 53 int *output_address = GetDeviceAddress<int>(outputs, 0); in Launch() local
86 … cudaMemcpyAsync(output_address, &output, sizeof(int), cudaMemcpyHostToDevice, in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_compiler.cc | 121 … auto output_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in CreateDeviceAddressForTensorValue() local
122 …if (output_address != nullptr && output_address->DeviceType() == device_context->GetDeviceAddressT… in CreateDeviceAddressForTensorValue()
|