Home
last modified time | relevance | path

Searched refs:PendingCounts (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dpending_counts_test.cc29 TEST(PendingCounts, Simple) { in TEST() argument
31 PendingCounts::Layout layout; in TEST()
32 std::vector<PendingCounts::Handle> h(C); in TEST()
37 PendingCounts c(layout); in TEST()
65 TEST(PendingCounts, CopyConstructor) { in TEST() argument
67 PendingCounts::Layout layout; in TEST()
68 std::vector<PendingCounts::Handle> h(C); in TEST()
72 PendingCounts c(layout); in TEST()
76 PendingCounts c2(c); in TEST()
83 TEST(PendingCounts, MarkLiveShowsUpAsCount) { in TEST() argument
[all …]
Dpropagator_state.h124 explicit IterationState(int64 iter_num, const PendingCounts* pending_counts, in IterationState()
150 int pending(PendingCounts::Handle h) { return counts.pending(h); } in pending()
151 int decrement_pending(PendingCounts::Handle h, int v) { in decrement_pending()
156 void mark_live(PendingCounts::Handle h) { counts.mark_live(h); } in mark_live()
158 void mark_started(PendingCounts::Handle h) { counts.mark_started(h); } in mark_started()
160 void mark_completed(PendingCounts::Handle h) { counts.mark_completed(h); } in mark_completed()
161 PendingCounts::NodeState node_state(PendingCounts::Handle h) { in node_state()
165 int dead_count(PendingCounts::Handle h) { return counts.dead_count(h); } in dead_count()
166 void increment_dead_count(PendingCounts::Handle h) { in increment_dead_count()
169 PendingCounts::AdjustResult adjust_for_activation(PendingCounts::Handle h, in adjust_for_activation()
[all …]
Dpending_counts.h50 class PendingCounts {
77 friend class PendingCounts;
83 explicit PendingCounts(Layout layout) in PendingCounts() function
92 explicit PendingCounts(const PendingCounts& other) in PendingCounts() function
100 ~PendingCounts() { delete[] bytes_; } in ~PendingCounts()
282 friend class PendingCounts;
372 void operator=(const PendingCounts&) = delete;
375 inline PendingCounts::Handle PendingCounts::Layout::CreateHandle( in CreateHandle()
Dimmutable_executor_state.h65 PendingCounts::Layout pending_counts_layout;
68 std::unique_ptr<PendingCounts> pending_counts;
90 const std::vector<PendingCounts::Handle>& pending_ids() const { in pending_ids()
135 std::vector<PendingCounts::Handle> pending_ids_;
Dpropagator_state.cc186 PendingCounts::Handle pending_id = in DumpIterationState()
188 if (iteration->node_state(pending_id) == PendingCounts::PENDING_NOTREADY || in DumpIterationState()
189 iteration->node_state(pending_id) == PendingCounts::PENDING_READY) { in DumpIterationState()
195 PendingCounts::Handle pending_id = in DumpIterationState()
197 if (iteration->node_state(pending_id) == PendingCounts::STARTED) { in DumpIterationState()
308 auto propagate_to_non_merge = [&](PendingCounts::Handle dst_pending_id) { in DeleteFrame()
416 const PendingCounts::Handle dst_pending_id = in ActivateNodesFastPathInternal()
428 const PendingCounts::AdjustResult adjust_result = in ActivateNodesFastPathInternal()
438 const PendingCounts::Handle dst_pending_id = in ActivateNodesFastPathInternal()
440 const PendingCounts::AdjustResult adjust_result = in ActivateNodesFastPathInternal()
[all …]
Dimmutable_executor_state.cc353 absl::make_unique<PendingCounts>(finfo->pending_counts_layout); in InitializePending()