| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_thunk.cc | 56 buffer_allocations.GetDeviceAddress(output_); in ExecuteOnStream()
57 se::DeviceMemoryBase operand = buffer_allocations.GetDeviceAddress(operand_); in ExecuteOnStream()
58 se::DeviceMemory<float> scale(buffer_allocations.GetDeviceAddress(scale_)); in ExecuteOnStream()
59 se::DeviceMemory<float> offset(buffer_allocations.GetDeviceAddress(offset_)); in ExecuteOnStream()
60 se::DeviceMemory<float> mean(buffer_allocations.GetDeviceAddress(mean_)); in ExecuteOnStream()
62 buffer_allocations.GetDeviceAddress(variance_)); in ExecuteOnStream()
92 se::DeviceMemoryBase operand = buffer_allocations.GetDeviceAddress(operand_); in ExecuteOnStream()
94 buffer_allocations.GetDeviceAddress(output_data_); in ExecuteOnStream()
97 buffer_allocations.GetDeviceAddress(output_mean_)); in ExecuteOnStream()
99 buffer_allocations.GetDeviceAddress(output_inv_stddev_)); in ExecuteOnStream()
[all …]
|
| D | fft_thunk.cc | 184 buffer_allocations.GetDeviceAddress(input_buffer_)); in ExecuteOnStream()
186 buffer_allocations.GetDeviceAddress(output_buffer_)); in ExecuteOnStream()
192 buffer_allocations.GetDeviceAddress(input_buffer_)); in ExecuteOnStream()
194 buffer_allocations.GetDeviceAddress(output_buffer_)); in ExecuteOnStream()
200 buffer_allocations.GetDeviceAddress(input_buffer_)); in ExecuteOnStream()
202 buffer_allocations.GetDeviceAddress(output_buffer_)); in ExecuteOnStream()
214 buffer_allocations.GetDeviceAddress(input_buffer_)); in ExecuteOnStream()
216 buffer_allocations.GetDeviceAddress(output_buffer_)); in ExecuteOnStream()
229 buffer_allocations.GetDeviceAddress(input_buffer_)); in ExecuteOnStream()
231 buffer_allocations.GetDeviceAddress(output_buffer_)); in ExecuteOnStream()
[all …]
|
| D | buffer_allocations.cc | 44 se::DeviceMemoryBase buffer_address = GetDeviceAddress(allocation.index()); in TearDown()
60 se::DeviceMemoryBase BufferAllocations::GetDeviceAddress( in GetDeviceAddress() function in xla::gpu::BufferAllocations
74 se::DeviceMemoryBase BufferAllocations::GetDeviceAddress( in GetDeviceAddress() function in xla::gpu::BufferAllocations
76 se::DeviceMemoryBase base = GetDeviceAddress(buffer_slice.index()); in GetDeviceAddress()
|
| D | convolution_thunk.cc | 48 operand_se_buffers.push_back(buffer_allocations.GetDeviceAddress(buffer)); in ExecuteOnStream()
52 buffer_allocations.GetDeviceAddress(result_buffer_); in ExecuteOnStream()
55 buffer_allocations.GetDeviceAddress(scratch_buffer_); in ExecuteOnStream()
|
| D | copy_thunk.cc | 34 params.buffer_allocations->GetDeviceAddress(destination_buffer_); in ExecuteOnStream()
51 params.buffer_allocations->GetDeviceAddress(destination_buffer_); in ExecuteOnStream()
53 params.buffer_allocations->GetDeviceAddress(source_buffer_); in ExecuteOnStream()
|
| D | nccl_all_to_all_thunk.cc | 83 params.buffer_allocations->GetDeviceAddress(buffer.source_buffer) in RunNcclCollective()
86 params.buffer_allocations->GetDeviceAddress(buffer.destination_buffer) in RunNcclCollective()
115 params.buffer_allocations->GetDeviceAddress(buffer.source_buffer) in RunNcclCollective()
118 params.buffer_allocations->GetDeviceAddress(buffer.destination_buffer) in RunNcclCollective()
|
| D | memset_thunk.cc | 26 params.buffer_allocations->GetDeviceAddress(dest_); in ExecuteOnStream()
35 params.buffer_allocations->GetDeviceAddress(dest_); in ExecuteOnStream()
|
| D | cholesky_thunk.cc | 71 params.buffer_allocations->GetDeviceAddress(a_buffer_).opaque()); in ExecuteOnStream()
73 params.buffer_allocations->GetDeviceAddress(info_buffer_).opaque()); in ExecuteOnStream()
75 params.buffer_allocations->GetDeviceAddress(workspace_buffer_); in ExecuteOnStream()
|
| D | buffer_allocations.h | 58 se::DeviceMemoryBase GetDeviceAddress(
67 se::DeviceMemoryBase GetDeviceAddress(
|
| D | nccl_all_gather_thunk.cc | 83 params.buffer_allocations->GetDeviceAddress(buffer.source_buffer) in RunNcclCollective()
86 params.buffer_allocations->GetDeviceAddress(buffer.destination_buffer) in RunNcclCollective()
|
| D | triangular_solve_thunk.cc | 86 buffer_allocations.GetDeviceAddress(a_buffer_).opaque()); in ExecuteOnStream()
88 buffer_allocations.GetDeviceAddress(b_buffer_).opaque()); in ExecuteOnStream()
|
| D | nccl_all_reduce_thunk.cc | 134 params.buffer_allocations->GetDeviceAddress(buffer.source_buffer) in RunNcclCollective()
137 params.buffer_allocations->GetDeviceAddress(buffer.destination_buffer) in RunNcclCollective()
|
| D | replica_id_thunk.cc | 25 auto dest_addr = params.buffer_allocations->GetDeviceAddress(dest_); in ExecuteOnStream()
|
| D | collective_permute_thunk.cc | 256 auto src_addr = params.buffer_allocations->GetDeviceAddress(src_); in ExecuteOnStream()
257 auto dest_addr = params.buffer_allocations->GetDeviceAddress(dest_); in ExecuteOnStream()
|
| D | custom_call_thunk.cc | 50 params.buffer_allocations->GetDeviceAddress(slice).opaque()); in ExecuteOnStream()
|
| D | infeed_thunk.cc | 57 buffer_allocations.GetDeviceAddress(dest_slices_[index++].slice); in ExecuteOnStream()
|
| D | while_thunk.cc | 59 params.buffer_allocations->GetDeviceAddress( in ExecuteOnStream()
|
| D | outfeed_thunk.cc | 90 buffer_allocations.GetDeviceAddress(source_slice); in ExecuteOnStream()
|
| D | conditional_thunk.cc | 79 params.buffer_allocations->GetDeviceAddress(branch_index_buffer_index_); in ExecuteOnStream()
|
| D | kernel_thunk.cc | 106 params.buffer_allocations->GetDeviceAddress(arg->index()); in ExecuteOnStream()
|
| D | gemm_thunk.cc | 60 return params.buffer_allocations->GetDeviceAddress(slice); in ExecuteOnStream()
|
| D | gpu_executable.cc | 538 buffer_allocations.GetDeviceAddress(output_info.allocation_index); in ExecuteAsyncOnStreamImpl()
|
| /external/libusb/libusb/os/ |
| D | darwin_usb.c | 1047 (*device)->GetDeviceAddress (device, (USBDeviceAddress *)&new_device->address); in darwin_get_cached_device()
|