Searched refs:input_dim_value_to_index_vector_ (Results 1 – 2 of 2) sorted by relevance
845 input_dim_value_to_index_vector_.push_back(-1); in OutputBatchIndexToInputIndex()847 input_dim_value_to_index_vector_.push_back( in OutputBatchIndexToInputIndex()917 if (input_dim_value_to_index_vector_[i] != -1) { in PropagateIndexVectorToInputIndex()918 input_index_[i] = index_vector_[input_dim_value_to_index_vector_[i]]; in PropagateIndexVectorToInputIndex()929 std::vector<int64> input_dim_value_to_index_vector_; member in xla::OutputBatchIndexToInputIndex
2119 input_dim_value_to_index_vector_.push_back(-1); in UpdateScatterIndexToInputIndex()2121 input_dim_value_to_index_vector_.push_back( in UpdateScatterIndexToInputIndex()2191 if (input_dim_value_to_index_vector_[i] != -1) { in PropagateIndexVectorToInputIndex()2192 input_index_[i] = index_vector_[input_dim_value_to_index_vector_[i]]; in PropagateIndexVectorToInputIndex()2203 std::vector<int64> input_dim_value_to_index_vector_; variable