Searched defs:a_buffer (Results 1 – 6 of 6) sorted by relevance
108 BufferAllocation::Slice a_buffer, in CholeskyThunk()131 se::DeviceMemoryBase a_buffer = in ExecuteOnStream() local
32 BufferAllocation::Slice a_buffer, BufferAllocation::Slice b_buffer, in CublasLtMatmulThunk()
73 se::DeviceMemoryBase a_buffer; member
37 const BufferAllocation::Slice& a_buffer, in TriangularSolveThunk()
687 Status MatmulPlan::DoMatmul(se::Stream* stream, se::DeviceMemoryBase a_buffer, in DoMatmul()718 se::Stream* stream, se::DeviceMemoryBase a_buffer, in ExecuteOnStream()
1319 se::DeviceMemoryBase a_buffer = GetDeviceAddress(a); in operator ()() local