Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.h420 const ShapeIndex& shape_index_a,
Dbuffer_assignment.cc448 const HloInstruction* hlo_a, const ShapeIndex& shape_index_a, in SharesSliceAtIndex() argument
450 return GetUniqueSlice(hlo_a, shape_index_a).ConsumeValueOrDie() == in SharesSliceAtIndex()