Home
last modified time | relevance | path

Searched refs:a_buffer (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcholesky_thunk.cc35 BufferAllocation::Slice a_buffer, in CholeskyThunk() argument
43 a_buffer_(a_buffer), in CholeskyThunk()
Dtriangular_solve_thunk.cc36 const BufferAllocation::Slice& a_buffer, in TriangularSolveThunk() argument
47 a_buffer_(a_buffer), in TriangularSolveThunk()
Dcholesky_thunk.h45 BufferAllocation::Slice a_buffer,
Dtriangular_solve_thunk.h42 const BufferAllocation::Slice& a_buffer,
Dir_emitter_unnested.cc500 auto a_buffer = assn.GetUniqueSlice(custom_call, {0}).ValueOrDie(); in HandleCustomCall() local
506 if (operand_buffer != a_buffer) { in HandleCustomCall()
509 /*destination_buffer=*/a_buffer, in HandleCustomCall()
514 options, a_buffer, workspace_buffer, info_buffer, in HandleCustomCall()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc754 const LogicalBuffer& a_buffer, in MayInterfereAcrossSubcomputations() argument
761 call_graph.NearestAncestorsInSameComputation(a_buffer.instruction(), in MayInterfereAcrossSubcomputations()
768 call_graph.InstructionIsNestedIn(a_buffer.instruction(), in MayInterfereAcrossSubcomputations()
774 VLOG(4) << "Can't interfere: " << a_buffer << " and " << b_buffer in MayInterfereAcrossSubcomputations()
781 VLOG(4) << "Can't interfere: " << a_buffer << " and " << b_buffer in MayInterfereAcrossSubcomputations()
811 const LogicalBuffer& a_buffer, in MaySkipInterferenceCheck() argument
813 if (a_buffer.instruction()->parent() == b_buffer.instruction()->parent()) { in MaySkipInterferenceCheck()
818 return !MayInterfereAcrossSubcomputations(assignment, a_buffer, b_buffer) || in MaySkipInterferenceCheck()
819 !MayInterfereAcrossSubcomputations(assignment, b_buffer, a_buffer); in MaySkipInterferenceCheck()