Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc888 int64 index_vector_index_i = 0; in PropagateOutputIndexGatherDimsToIndexVectorIndex() local
894 if (index_vector_index_i == dim_numbers_.index_vector_dim()) { in PropagateOutputIndexGatherDimsToIndexVectorIndex()
895 index_vector_index_i++; in PropagateOutputIndexGatherDimsToIndexVectorIndex()
898 index_vector_index_[index_vector_index_i++] = output_index[i]; in PropagateOutputIndexGatherDimsToIndexVectorIndex()
Dhlo_evaluator_typed_visitor.h2162 int64 index_vector_index_i = 0; in PropagateUpdateIndexScatterDimsToIndexVectorIndex() local
2168 if (index_vector_index_i == dim_numbers_.index_vector_dim()) { in PropagateUpdateIndexScatterDimsToIndexVectorIndex()
2169 index_vector_index_i++; in PropagateUpdateIndexScatterDimsToIndexVectorIndex()
2172 index_vector_index_[index_vector_index_i++] = update_index[i]; in PropagateUpdateIndexScatterDimsToIndexVectorIndex()