Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_expander.cc110 static StatusOr<HloInstruction*> ExpandIndexVectorIntoOperandSpace( in ExpandIndexVectorIntoOperandSpace() function
197 ExpandIndexVectorIntoOperandSpace(index_vector, dim_numbers, in GatherLoopBody()
Dscatter_expander.cc132 static StatusOr<HloInstruction*> ExpandIndexVectorIntoOperandSpace( in ExpandIndexVectorIntoOperandSpace() function
258 ExpandIndexVectorIntoOperandSpace(index_vector, dim_numbers, in ScatterLoopBody()