Searched defs:DoMatmul (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ | ||
| H A D | cuda_blas_lt.cc | 257 port::Status BlasLt::DoMatmul(Stream* stream, const BlasLt::MatmulPlan& plan, in DoMatmul() function in stream_executor::cuda::BlasLt |
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ | ||
| H A D | matmul_utils.cc | 687 Status MatmulPlan::DoMatmul(se::Stream* stream, se::DeviceMemoryBase a_buffer, in DoMatmul() function in xla::gpu::cublas_lt::MatmulPlan |