Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcholesky_thunk.cc35 BufferAllocation::Slice a_buffer, in CholeskyThunk()
Dtriangular_solve_thunk.cc36 const BufferAllocation::Slice& a_buffer, in TriangularSolveThunk()
Dir_emitter_unnested.cc500 auto a_buffer = assn.GetUniqueSlice(custom_call, {0}).ValueOrDie(); in HandleCustomCall() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc754 const LogicalBuffer& a_buffer, in MayInterfereAcrossSubcomputations()
811 const LogicalBuffer& a_buffer, in MaySkipInterferenceCheck()