Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_expander.cc29 static StatusOr<HloInstruction*> TransposeIndexVectorDimToLast( in TransposeIndexVectorDimToLast() function
61 TransposeIndexVectorDimToLast(start_indices, index_vector_dim)); in CanonicalizeGatherIndices()
Dscatter_expander.cc31 static StatusOr<HloInstruction*> TransposeIndexVectorDimToLast( in TransposeIndexVectorDimToLast() function
61 TransposeIndexVectorDimToLast(scatter_indices, index_vector_dim)); in CanonicalizeScatterIndices()