Searched refs:PropagateOutputIndexGatherDimsToIndexVectorIndex (Results 1 – 1 of 1) sorted by relevance
1544 PropagateOutputIndexGatherDimsToIndexVectorIndex(output_index); in operator ()()1555 void PropagateOutputIndexGatherDimsToIndexVectorIndex( in PropagateOutputIndexGatherDimsToIndexVectorIndex() function in xla::OutputBatchIndexToInputIndex