Searched refs:index_vector_ (Results 1 – 2 of 2) sorted by relevance
856 index_vector_.resize(index_vector_size); in OutputBatchIndexToInputIndex()906 for (int64 i = 0, e = index_vector_.size(); i < e; i++) { in FetchIndexVector()908 TF_ASSIGN_OR_RETURN(index_vector_[i], in FetchIndexVector()918 input_index_[i] = index_vector_[input_dim_value_to_index_vector_[i]]; in PropagateIndexVectorToInputIndex()940 std::vector<int64> index_vector_; member in xla::OutputBatchIndexToInputIndex
2130 index_vector_.resize(index_vector_size); in UpdateScatterIndexToInputIndex()2180 for (int64 i = 0, e = index_vector_.size(); i < e; i++) { in FetchIndexVector()2182 TF_ASSIGN_OR_RETURN(index_vector_[i], scatter_indices_.GetIntegralAsS64( in FetchIndexVector()2192 input_index_[i] = index_vector_[input_dim_value_to_index_vector_[i]]; in PropagateIndexVectorToInputIndex()2214 std::vector<int64> index_vector_; variable