Searched refs:rhs_buffer (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.h | 53 const BufferAllocation::Slice& rhs_buffer,
81 se::DeviceMemoryBase rhs_buffer, se::DeviceMemoryBase output_buffer,
|
| D | buffer_comparator_test.cc | 47 se::ScopedDeviceMemory<ElementType> rhs_buffer = in CompareEqualBuffers() local
51 stream.ThenMemcpy(rhs_buffer.ptr(), rhs.data(), rhs_buffer->size()); in CompareEqualBuffers()
59 return comparator.CompareEqual(&stream, *lhs_buffer, *rhs_buffer) in CompareEqualBuffers()
|
| D | gemm_thunk.cc | 48 const BufferAllocation::Slice &rhs_buffer, in GemmThunk() argument
54 rhs_buffer_(rhs_buffer), in GemmThunk()
172 se::DeviceMemoryBase lhs_buffer, se::DeviceMemoryBase rhs_buffer, in RunGemm() argument
246 rhs_buffer, rhs_shape, dim_nums.rhs_contracting_dimensions(0) == col_dim); in RunGemm()
|
| D | gemm_algorithm_picker.cc | 90 TF_ASSIGN_OR_RETURN(se::DeviceMemoryBase rhs_buffer, in DoUncachedGemmAutotune()
134 CHECK(RunGemm(config, lhs_buffer, rhs_buffer, output_buffer, stream, in DoUncachedGemmAutotune()
|