Searched refs:GetMutablePtr (Results 1 – 12 of 12) sorted by relevance
118 io_addrs.emplace_back(reinterpret_cast<uintptr_t>(input_addr->GetMutablePtr())); in UpdateInputOutputAddr()124 io_addrs.emplace_back(reinterpret_cast<uintptr_t>(output_addr->GetMutablePtr())); in UpdateInputOutputAddr()
124 … [](const DeviceAddressPtr &address_ptr) { return address_ptr->GetMutablePtr(); }); in UpdateArgs()
46 virtual void *GetMutablePtr() const = 0;
381 launch_info_.inputs_[i]->addr = input_device_tensors_[i]->GetMutablePtr(); in PreLaunchKernel()388 launch_info_.outputs_[i]->addr = output_device_tensors_[i]->GetMutablePtr(); in PreLaunchKernel()395 launch_info_.workspaces_[i]->addr = workspace_device_tensors_[i]->GetMutablePtr(); in PreLaunchKernel()
162 return src_device_tensor->SyncDeviceToHost(copy_size, dst_device_tensor->GetMutablePtr()); in Copy()
164 launch_info_.outputs_[i]->addr = device_tensors[i]->GetMutablePtr(); in OnMemoryAllocFinish()
79 auto origin_dev_ptr = device_address->GetMutablePtr(); in UpdateTensorAddr()
85 void *GetMutablePtr() const override { return ptr_; } in GetMutablePtr() function
191 void *current_step = current_step_tensor->device_address()->GetMutablePtr(); in SetOpMappingInfo()192 void *current_epoch = currnet_epoch_tensor->device_address()->GetMutablePtr(); in SetOpMappingInfo()193 void *steps_per_epoch = steps_per_epoch_tensor->device_address()->GetMutablePtr(); in SetOpMappingInfo()
67 …std::make_shared<kernel::Address>(static_cast<uint8_t *>(device_address->GetMutablePtr()), origin_… in AllocateAllReduceAddr()
52 static_cast<uint8_t *>(device_address->GetMutablePtr()), device_address->GetSize())); in AllocateAllReduceAddr()
743 auto old_ptr = old_device_address->GetMutablePtr(); in UpdateTensorInfo()