Searched refs:lhs_buffer (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_comparator_test.cc | 45 se::ScopedDeviceMemory<ElementType> lhs_buffer = in CompareEqualBuffers() local
50 stream.ThenMemcpy(lhs_buffer.ptr(), lhs.data(), lhs_buffer->size()); in CompareEqualBuffers()
57 {static_cast<int64>(lhs_buffer->ElementCount())}), in CompareEqualBuffers()
59 return comparator.CompareEqual(&stream, *lhs_buffer, *rhs_buffer) in CompareEqualBuffers()
|
| D | gemm_thunk.h | 52 const BufferAllocation::Slice& lhs_buffer,
80 const GpuGemmConfig& gemm_config, se::DeviceMemoryBase lhs_buffer,
|
| D | gemm_thunk.cc | 47 const BufferAllocation::Slice &lhs_buffer, in GemmThunk() argument
53 lhs_buffer_(lhs_buffer), in GemmThunk()
172 se::DeviceMemoryBase lhs_buffer, se::DeviceMemoryBase rhs_buffer, in RunGemm() argument
244 lhs_buffer, lhs_shape, dim_nums.lhs_contracting_dimensions(0) == row_dim); in RunGemm()
|
| D | gemm_algorithm_picker.cc | 88 TF_ASSIGN_OR_RETURN(se::DeviceMemoryBase lhs_buffer, in DoUncachedGemmAutotune()
134 CHECK(RunGemm(config, lhs_buffer, rhs_buffer, output_buffer, stream, in DoUncachedGemmAutotune()
|