Searched defs:device_sync (Results 1 – 9 of 9) sorted by relevance
139 MapTensor::ExportData MapTensor::ExportDataFromDevice(const DeviceSyncPtr &device_sync, bool increm… in ExportDataFromDevice()205 DeviceSyncPtr device_sync = device_address(); in Export() local223 DeviceSyncPtr device_sync = device_address(); in ExportSlice() local
296 void BaseTensor::set_device_address(const DeviceSyncPtr &device_sync, bool need_update_ref_count) { in set_device_address()
287 void Tensor::data_sync_directly(const DeviceSync *const device_sync, bool need_wait) const { in data_sync_directly() argument
680 auto device_sync = input_tensor->device_address(); in UpdateInputTensorForHeterogeneous() local950 const auto &device_sync = input_tensor->device_address(); in UpdateInputDeviceAddress() local992 const auto &device_sync = input_tensor->device_address(); in CopyHostToDevice() local
76 auto device_sync = tensor->device_address(); in ClearDeviceMemory() local
173 DeviceSyncPtr PyBoostUtils::ContiguousByDeviceAddress(const DeviceSyncPtr &device_sync) { in ContiguousByDeviceAddress()
938 const auto &device_sync = input_tensor->device_address(); in CreateInputAddressForViewOp() local
893 const auto &device_sync = tensor->device_address(); in MallocForInput() local
115 auto device_sync = tensor->device_address(); in SyncDeviceInfoToValueNode() local