Searched refs:GetDeviceAddress (Results 1 – 19 of 19) sorted by relevance
109 se::DeviceMemory<float> output(buffer_allocations.GetDeviceAddress(output_)); in ExecuteOnStream()112 se::DeviceMemory<float>(buffer_allocations.GetDeviceAddress(operand_)), in ExecuteOnStream()113 se::DeviceMemory<float>(buffer_allocations.GetDeviceAddress(scale_)), in ExecuteOnStream()114 se::DeviceMemory<float>(buffer_allocations.GetDeviceAddress(offset_)), in ExecuteOnStream()115 se::DeviceMemory<float>(buffer_allocations.GetDeviceAddress(mean_)), in ExecuteOnStream()116 se::DeviceMemory<float>(buffer_allocations.GetDeviceAddress(variance_)), in ExecuteOnStream()175 buffer_allocations.GetDeviceAddress(output_data_)); in ExecuteOnStream()177 buffer_allocations.GetDeviceAddress(output_mean_)); in ExecuteOnStream()179 buffer_allocations.GetDeviceAddress(output_inv_stddev_)); in ExecuteOnStream()184 se::DeviceMemory<float>(buffer_allocations.GetDeviceAddress(operand_)), in ExecuteOnStream()[all …]
161 buffer_allocations.GetDeviceAddress(input_buffer_)); in ExecuteOnStream()163 buffer_allocations.GetDeviceAddress(output_buffer_)); in ExecuteOnStream()170 buffer_allocations.GetDeviceAddress(input_buffer_)); in ExecuteOnStream()172 buffer_allocations.GetDeviceAddress(output_buffer_)); in ExecuteOnStream()186 buffer_allocations.GetDeviceAddress(input_buffer_)); in ExecuteOnStream()188 buffer_allocations.GetDeviceAddress(output_buffer_)); in ExecuteOnStream()195 buffer_allocations.GetDeviceAddress(input_buffer_)); in ExecuteOnStream()197 buffer_allocations.GetDeviceAddress(output_buffer_)); in ExecuteOnStream()
49 operand_se_buffers.push_back(buffer_allocations.GetDeviceAddress(buffer)); in ExecuteOnStream()53 buffer_allocations.GetDeviceAddress(result_buffer_); in ExecuteOnStream()56 buffer_allocations.GetDeviceAddress(scratch_buffer_); in ExecuteOnStream()65 buffer_allocations.GetDeviceAddress(tuple_result_buffer_)); in ExecuteOnStream()
67 buffer_allocations.GetDeviceAddress(tuple_element_buffer); in ExecuteOnStream()90 buffer_allocations.GetDeviceAddress( in ExecuteOnStream()105 buffer_allocations.GetDeviceAddress(infeed_slices_.element({0})); in ExecuteOnStream()108 buffer_allocations.GetDeviceAddress(infeed_slices_.element({})); in ExecuteOnStream()
37 buffer_allocations.GetDeviceAddress(destination_buffer_); in ExecuteOnStream()56 buffer_allocations.GetDeviceAddress(destination_buffer_); in ExecuteOnStream()58 buffer_allocations.GetDeviceAddress(source_buffer_); in ExecuteOnStream()
136 se::DeviceMemoryBase buffer_address = GetDeviceAddress(allocation.index()); in TearDown()153 se::DeviceMemoryBase BufferAllocations::GetDeviceAddress( in GetDeviceAddress() function in xla::gpu::BufferAllocations160 se::DeviceMemoryBase BufferAllocations::GetDeviceAddress( in GetDeviceAddress() function in xla::gpu::BufferAllocations162 se::DeviceMemoryBase base = GetDeviceAddress(buffer_slice.index()); in GetDeviceAddress()
75 buffer_allocations.GetDeviceAddress(a_buffer_).opaque()); in ExecuteOnStream()77 buffer_allocations.GetDeviceAddress(info_buffer_).opaque()); in ExecuteOnStream()79 buffer_allocations.GetDeviceAddress(workspace_buffer_); in ExecuteOnStream()
27 se::DeviceMemoryBase dest_data = buffer_allocations.GetDeviceAddress(dest_); in ExecuteOnStream()36 se::DeviceMemoryBase dest_data = buffer_allocations.GetDeviceAddress(dest_); in ExecuteOnStream()
32 buffer_allocations.GetDeviceAddress(tuple_element_buffers_[i]).opaque(); in ExecuteOnStream()35 buffer_allocations.GetDeviceAddress(dest_buffer_)); in ExecuteOnStream()
71 se::DeviceMemoryBase GetDeviceAddress(76 se::DeviceMemoryBase GetDeviceAddress(
59 data_address = buffer_allocations.GetDeviceAddress(slice); in ExecuteOnStream()71 buffer_allocations.GetDeviceAddress(tuple_slice); in ExecuteOnStream()
85 buffer_allocations.GetDeviceAddress(a_buffer_).opaque()); in ExecuteOnStream()87 buffer_allocations.GetDeviceAddress(b_buffer_).opaque()); in ExecuteOnStream()
324 participant.source_data = buffer_allocations.GetDeviceAddress(source_buffer_); in ExecuteOnStream()326 buffer_allocations.GetDeviceAddress(destination_buffer_); in ExecuteOnStream()
317 buffer_allocations.GetDeviceAddress(lhs_buffer_); in ExecuteOnStream()319 buffer_allocations.GetDeviceAddress(rhs_buffer_); in ExecuteOnStream()321 buffer_allocations.GetDeviceAddress(output_buffer_); in ExecuteOnStream()
54 buffer_allocations.GetDeviceAddress(condition_result_buffer_index_); in ExecuteOnStream()
67 buffer_allocations.GetDeviceAddress(branch_index_buffer_index_); in ExecuteOnStream()
96 const auto& buf = buffer_allocations.GetDeviceAddress(arg->index()); in ExecuteOnStream()
315 buffer_allocations->GetDeviceAddress(slice.index()); in ExecuteOnStream()
927 (*device)->GetDeviceAddress (device, (USBDeviceAddress *)&new_device->address); in darwin_get_cached_device()