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