Home
last modified time | relevance | path

Searched refs:a_buffer_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcholesky_thunk.cc43 a_buffer_(a_buffer), in CholeskyThunk()
59 << " a=" << a_buffer_.ToString() in ExecuteOnStream()
75 buffer_allocations.GetDeviceAddress(a_buffer_).opaque()); in ExecuteOnStream()
Dtriangular_solve_thunk.cc47 a_buffer_(a_buffer), in TriangularSolveThunk()
85 buffer_allocations.GetDeviceAddress(a_buffer_).opaque()); in ExecuteOnStream()
Dtriangular_solve_thunk.h61 const BufferAllocation::Slice a_buffer_; variable
Dcholesky_thunk.h61 const BufferAllocation::Slice a_buffer_; variable