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