Searched refs:index_vector_size (Results 1 – 5 of 5) sorted by relevance
185 int64 index_vector_size = start_indices->shape().dimensions(1); in GatherLoopBody() local189 {1, index_vector_size})); in GatherLoopBody()
248 int index_vector_size = scatter_indices->shape().dimensions(1); in ScatterLoopBody() local252 {1, index_vector_size})); in ScatterLoopBody()
854 int64 index_vector_size = in OutputBatchIndexToInputIndex() local856 index_vector_.resize(index_vector_size); in OutputBatchIndexToInputIndex()
1948 int64 index_vector_size = in EmitElementalGather() local1950 for (int64 i = 0; i < index_vector_size; i++) { in EmitElementalGather()
2128 int64 index_vector_size = in UpdateScatterIndexToInputIndex() local2130 index_vector_.resize(index_vector_size); in UpdateScatterIndexToInputIndex()