Home
last modified time | relevance | path

Searched refs:hlo_b (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.h421 const HloInstruction* hlo_b,
427 const HloInstruction* hlo_b) const { in SharesTopLevelSlice() argument
428 return SharesSliceAtIndex(hlo_a, {}, hlo_b, {}); in SharesTopLevelSlice()
435 const HloInstruction* hlo_b) const;
Dbuffer_assignment.cc449 const HloInstruction* hlo_b, const ShapeIndex& shape_index_b) const { in SharesSliceAtIndex() argument
451 GetUniqueSlice(hlo_b, shape_index_b).ConsumeValueOrDie(); in SharesSliceAtIndex()
455 const HloInstruction* hlo_b) const { in HaveDisjointSlices()
478 SliceSet slices_b = collect_slices(hlo_b); in HaveDisjointSlices()