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()
1523 int64 index_vector_size = in OutputBatchIndexToInputIndex() local1525 index_vector_.resize(index_vector_size); in OutputBatchIndexToInputIndex()
2151 int64 index_vector_size =2153 index_vector_.resize(index_vector_size);
2007 int64 index_vector_size = in EmitElementalGather() local2009 for (int64 i = 0; i < index_vector_size; i++) { in EmitElementalGather()