Searched refs:hlo_a (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment.h | 439 bool SharesSliceAtIndex(const HloInstruction* hlo_a,
446 bool SharesTopLevelSlice(const HloInstruction* hlo_a, in SharesTopLevelSlice() argument
448 return SharesSliceAtIndex(hlo_a, {}, hlo_b, {}); in SharesTopLevelSlice()
454 bool HaveDisjointSlices(const HloInstruction* hlo_a,
|
| D | buffer_assignment.cc | 510 const HloInstruction* hlo_a, const ShapeIndex& shape_index_a, in SharesSliceAtIndex() argument
512 return GetUniqueSlice(hlo_a, shape_index_a).ConsumeValueOrDie() == in SharesSliceAtIndex()
516 bool BufferAssignment::HaveDisjointSlices(const HloInstruction* hlo_a, in HaveDisjointSlices() argument
539 SliceSet slices_a = collect_slices(hlo_a); in HaveDisjointSlices()
|