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.h440 const ShapeIndex& shape_index_a,
Dbuffer_assignment.cc510 const HloInstruction* hlo_a, const ShapeIndex& shape_index_a, in SharesSliceAtIndex() argument
512 return GetUniqueSlice(hlo_a, shape_index_a).ConsumeValueOrDie() == in SharesSliceAtIndex()