Home
last modified time | relevance | path

Searched defs:rhs_buffer (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgemm_thunk.cc31 const BufferAllocation::Slice& rhs_buffer, in GemmThunk()
H A Dbuffer_comparator_test.cc47 se::ScopedDeviceMemory<ElementType> rhs_buffer = in CompareEqualBuffers() local
H A Dgemm_algorithm_picker.cc96 se::DeviceMemoryBase lhs_buffer, se::DeviceMemoryBase rhs_buffer, in GetBestAlgorithm()
H A Dmatmul_utils.cc487 se::DeviceMemoryBase rhs_buffer, in RunGemm()