Searched refs:input_dim_value_to_index_vector_ (Results 1 – 2 of 2) sorted by relevance
1514 input_dim_value_to_index_vector_.push_back(-1); in OutputBatchIndexToInputIndex()1516 input_dim_value_to_index_vector_.push_back( in OutputBatchIndexToInputIndex()1587 if (input_dim_value_to_index_vector_[i] != -1) { in PropagateIndexVectorToInputIndex()1588 input_index_[i] = index_vector_[input_dim_value_to_index_vector_[i]]; in PropagateIndexVectorToInputIndex()1599 std::vector<int64> input_dim_value_to_index_vector_; member in xla::OutputBatchIndexToInputIndex
2142 input_dim_value_to_index_vector_.push_back(-1);2144 input_dim_value_to_index_vector_.push_back(2214 if (input_dim_value_to_index_vector_[i] != -1) {2215 input_index_[i] = index_vector_[input_dim_value_to_index_vector_[i]];2226 std::vector<int64> input_dim_value_to_index_vector_;