Searched defs:IterationState (Results 1 – 1 of 1) sorted by relevance
123 struct IterationState { struct124 explicit IterationState(int64 iter_num, const PendingCounts* pending_counts, in IterationState() function133 const int64 iter_num; // The index of this iteration in the enclosing loop.143 Entry* input_tensors;146 std::atomic<size_t> outstanding_ops;149 int outstanding_frame_count;150 int pending(PendingCounts::Handle h) { return counts.pending(h); } in pending()151 int decrement_pending(PendingCounts::Handle h, int v) { in decrement_pending()178 ~IterationState() { delete[] input_tensors; } in ~IterationState() argument