Searched defs:output_address (Results 1 – 5 of 5) sorted by relevance
48 bool KernelInfo::SetOutputAddr(const DeviceAddressPtr &output_address, size_t index) { in SetOutputAddr()91 bool KernelInfo::SetWorkspaceAddr(const DeviceAddressPtr &output_address, size_t index) { in SetWorkspaceAddr()
256 …auto output_address = std::dynamic_pointer_cast<device::DeviceAddress>(current_tensor->device_addr… in RunOpMallocPre() local407 …auto output_address = std::dynamic_pointer_cast<device::DeviceAddress>(current_tensor->device_addr… in RunOpAssignInputMemory() local965 … auto output_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in AssignValueNodeTensor() local
49 for (auto &output_address : kernel_info_->output_address_list()) { in Init() local355 auto output_address = output_addresses[i].get(); in FetchOutputDeviceTensor() local
53 int *output_address = GetDeviceAddress<int>(outputs, 0); in Launch() local
121 … auto output_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in CreateDeviceAddressForTensorValue() local