Searched refs:input_index_ (Results 1 – 3 of 3) sorted by relevance
60 input_index_(std::numeric_limits<int64>::min()) { in Barrier()139 if (new_elements) ++input_index_; in TryInsertMany()144 ctx, input_index_ != std::numeric_limits<int64>::max(), in TryInsertMany()146 "Barrier has had ", input_index_, in TryInsertMany()349 allocate_index_tensor->scalar<int64>()() = input_index_; in InsertOneLocked()435 int64 input_index_ TF_GUARDED_BY(mu_);
1522 input_index_.resize(input_shape.dimensions_size()); in OutputBatchIndexToInputIndex()1547 return absl::Span<const int64>(input_index_); in operator ()()1586 for (int64 i = 0, e = input_index_.size(); i < e; i++) { in PropagateIndexVectorToInputIndex()1588 input_index_[i] = index_vector_[input_dim_value_to_index_vector_[i]]; in PropagateIndexVectorToInputIndex()1614 std::vector<int64> input_index_; member in xla::OutputBatchIndexToInputIndex1650 input_index_.resize(input_shape.dimensions_size()); in OutputOffsetIndexToInputIndex()1665 return absl::Span<const int64>(input_index_); in operator ()()1679 for (int64 i = 0, e = input_index_.size(); i < e; i++) { in PropagateOutputIndexWindowDimsToInputIndex()1681 input_index_[i] = output_index[input_dim_value_to_output_index_[i]]; in PropagateOutputIndexWindowDimsToInputIndex()1696 std::vector<int64> input_index_; member in xla::OutputOffsetIndexToInputIndex
2150 input_index_.resize(input_shape.dimensions_size());2175 return absl::Span<const int64>(input_index_);2213 for (int64 i = 0, e = input_index_.size(); i < e; i++) {2215 input_index_[i] = index_vector_[input_dim_value_to_index_vector_[i]];2241 std::vector<int64> input_index_;2281 input_index_.resize(input_shape.dimensions_size());2296 return absl::Span<const int64>(input_index_);2310 for (int64 i = 0, e = input_index_.size(); i < e; i++) {2312 input_index_[i] = update_index[input_dim_value_to_update_index_[i]];2327 std::vector<int64> input_index_;