Searched refs:input_index_ (Results 1 – 4 of 4) 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_ GUARDED_BY(mu_);
853 input_index_.resize(input_shape.dimensions_size()); in OutputBatchIndexToInputIndex()878 return absl::Span<const int64>(input_index_); in operator ()()916 for (int64 i = 0, e = input_index_.size(); i < e; i++) { in PropagateIndexVectorToInputIndex()918 input_index_[i] = index_vector_[input_dim_value_to_index_vector_[i]]; in PropagateIndexVectorToInputIndex()944 std::vector<int64> input_index_; member in xla::OutputBatchIndexToInputIndex980 input_index_.resize(input_shape.dimensions_size()); in OutputOffsetIndexToInputIndex()995 return absl::Span<const int64>(input_index_); in operator ()()1009 for (int64 i = 0, e = input_index_.size(); i < e; i++) { in PropagateOutputIndexWindowDimsToInputIndex()1011 input_index_[i] = output_index[input_dim_value_to_output_index_[i]]; in PropagateOutputIndexWindowDimsToInputIndex()1026 std::vector<int64> input_index_; member in xla::OutputOffsetIndexToInputIndex
2127 input_index_.resize(input_shape.dimensions_size()); in UpdateScatterIndexToInputIndex()2152 return absl::Span<const int64>(input_index_); in operator()2190 for (int64 i = 0, e = input_index_.size(); i < e; i++) { in PropagateIndexVectorToInputIndex()2192 input_index_[i] = index_vector_[input_dim_value_to_index_vector_[i]]; in PropagateIndexVectorToInputIndex()2218 std::vector<int64> input_index_; variable2258 input_index_.resize(input_shape.dimensions_size()); in UpdateWindowIndexToInputIndex()2273 return absl::Span<const int64>(input_index_); in operator()2287 for (int64 i = 0, e = input_index_.size(); i < e; i++) { in PropagateUpdateIndexWindowDimsToInputIndex()2289 input_index_[i] = update_index[input_dim_value_to_update_index_[i]]; in PropagateUpdateIndexWindowDimsToInputIndex()2304 std::vector<int64> input_index_; variable
177 ++input_index_;183 ++input_index_;200 int input_index() const { return input_index_; } in input_index()206 : position_(position), input_index_(0) {} in iterator()209 int input_index_; variable