Searched refs:a_buffer (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cholesky_thunk.cc | 35 BufferAllocation::Slice a_buffer, in CholeskyThunk() argument
43 a_buffer_(a_buffer), in CholeskyThunk()
|
| D | triangular_solve_thunk.cc | 36 const BufferAllocation::Slice& a_buffer, in TriangularSolveThunk() argument
47 a_buffer_(a_buffer), in TriangularSolveThunk()
|
| D | cholesky_thunk.h | 45 BufferAllocation::Slice a_buffer,
|
| D | triangular_solve_thunk.h | 42 const BufferAllocation::Slice& a_buffer,
|
| D | ir_emitter_unnested.cc | 500 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/ |
| D | buffer_assignment.cc | 754 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()
|