| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | propagator_state.h | 63 const NodeItem* node_item;
69 TaggedNode(const NodeItem* node_item, FrameState* in_frame, in TaggedNode()
76 const NodeItem& get_node_item() const { return *node_item; } in get_node_item()
264 std::vector<std::pair<const NodeItem*, Entry>> next_iter_roots
272 std::vector<std::pair<const NodeItem*, Entry>> inv_values TF_GUARDED_BY(mu);
276 std::vector<const NodeItem*> dead_exits TF_GUARDED_BY(mu);
281 std::vector<const NodeItem*>* nodes = nullptr;
347 void AddLoopInv(const NodeItem* item, const Entry& entry,
358 bool ActivateNodesAndAdjustOutstanding(const NodeItem* item,
365 int ActivateNodesLocked(const NodeItem* item, const bool is_dead,
[all …]
|
| D | graph_view.h | 59 struct NodeItem { struct
150 NodeItem() {} in NodeItem() argument
164 sizeof(NodeItem)); in var() argument
201 TF_DISALLOW_COPY_AND_ASSIGN(NodeItem); argument
218 NodeItem* node(int32 id) const { in node()
224 : reinterpret_cast<NodeItem*>(space_ + node_offsets_[id])); in node()
230 const NodeItem& node_ref(int32 id) const { in node_ref()
235 return *reinterpret_cast<NodeItem*>(space_ + node_offsets_[id]); in node_ref()
|
| D | graph_view.cc | 38 string NodeItem::DebugString() const { in DebugString()
54 NodeItem* n = node(i); in ~GraphView()
56 n->NodeItem::~NodeItem(); in ~GraphView()
94 sizeof(NodeItem) // Fixed in NodeItemBytes()
102 static constexpr size_t kItemAlignment = sizeof(NodeItem*); in NodeItemBytes()
103 static_assert(kItemAlignment % alignof(NodeItem) == 0, in NodeItemBytes()
111 static_assert(sizeof(NodeItem) % alignof(EdgeInfo) == 0, in NodeItemBytes()
113 static_assert(sizeof(NodeItem) % alignof(AllocatorAttributes) == 0, in NodeItemBytes()
127 constexpr size_t kItemAlignment = sizeof(NodeItem*); in InitializeNode()
129 NodeItem* item = reinterpret_cast<NodeItem*>(ptr); in InitializeNode()
[all …]
|
| D | propagator_state.cc | 54 void PropagatorState::ActivateRoots(gtl::ArraySlice<const NodeItem*> roots, in ActivateRoots()
58 for (const NodeItem* item : roots) { in ActivateRoots()
79 const NodeItem* const item = tagged_node.node_item; in PropagateOutputs()
183 const std::vector<const NodeItem*>* nodes = frame->nodes; in DumpIterationState()
185 for (const NodeItem* node : *nodes) { in DumpIterationState()
194 for (const NodeItem* node : *nodes) { in DumpIterationState()
231 const NodeItem& node_item, in FindOrCreateChildFrame()
297 for (const NodeItem* item : frame->dead_exits) { in DeleteFrame()
298 auto maybe_add_to_ready = [&](const NodeItem& dst_item, bool dst_ready, in DeleteFrame()
314 const NodeItem& dst_item = in DeleteFrame()
[all …]
|
| D | simple_propagator_state.h | 56 const NodeItem* node_item;
58 explicit TaggedNode(const NodeItem* node_item) : node_item(node_item) {} in TaggedNode()
60 const NodeItem& get_node_item() const { return *node_item; } in get_node_item()
109 void ActivateRoots(gtl::ArraySlice<const NodeItem*> roots,
184 const std::vector<const NodeItem*>* const nodes_;
|
| D | propagator_debug_utils.h | 21 struct NodeItem;
29 void DumpPendingNodeState(const NodeItem& node_item, const Entry* input_vector,
34 void DumpActiveNodeState(const NodeItem& node_item, const Entry* input_vector);
|
| D | simple_propagator_state.cc | 49 gtl::ArraySlice<const NodeItem*> roots, TaggedNodeSeq* ready) { in ActivateRoots()
50 for (const NodeItem* item : roots) { in ActivateRoots()
75 const NodeItem* item = tagged_node.node_item; in PropagateOutputs()
109 for (const NodeItem* node : *nodes_) { in DumpState()
115 for (const NodeItem* node : *nodes_) { in DumpState()
|
| D | immutable_executor_state.h | 71 std::unique_ptr<std::vector<const NodeItem*>> nodes;
93 const std::vector<const NodeItem*>& root_nodes() const { return root_nodes_; } in root_nodes()
97 const FrameInfo& get_enter_frame_info(const NodeItem& node_item) const { in get_enter_frame_info()
138 std::vector<const NodeItem*> root_nodes_;
|
| D | executor.cc | 174 bool IsExpensive(const NodeItem& node) const { in IsExpensive()
181 bool HasExpensiveMarker(const NodeItem& node) const { in HasExpensiveMarker()
189 void UpdateCostEstimate(const NodeItem& node, uint64 elapsed_cycles) { in UpdateCostEstimate()
292 Status ProcessSync(const NodeItem& item, OpKernelContext::Params* params,
294 void ProcessAsync(const NodeItem& item, const OpKernelContext::Params& params,
298 void ProcessConstTensor(const NodeItem& item, EntryVector* outputs,
302 Status PrepareInputs(const NodeItem& item, Entry* first_input,
308 Status ProcessOutputs(const NodeItem& item, OpKernelContext* ctx,
493 const NodeItem* _item, Entry* _first_input, in AsyncState()
513 const NodeItem* item;
[all …]
|
| D | immutable_executor_state.cc | 38 NodeItem* item = gview_.node(i); in ~ImmutableExecutorState()
93 absl::make_unique<std::vector<const NodeItem*>>(); in Initialize()
126 NodeItem* item = gview_.node(id); in Initialize()
248 NodeItem* item = gview_.node(id); in Initialize()
252 NodeItem* dst_item = gview_.node(dst_id); in Initialize()
|
| D | propagator_debug_utils.cc | 43 void DumpPendingNodeState(const NodeItem& node_item, const Entry* input_vector, in DumpPendingNodeState()
75 void DumpActiveNodeState(const NodeItem& node_item, const Entry* input_vector) { in DumpActiveNodeState()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | eager_executor.h | 185 struct NodeItem : core::RefCounted { struct
196 void NodeDone(const core::RefCountPtr<NodeItem>& item, const Status& status, argument
206 Status RunItem(core::RefCountPtr<NodeItem> item, bool from_queue);
207 Status MoveToUnfinished(core::RefCountPtr<NodeItem> item, bool from_queue);
224 std::queue<core::RefCountPtr<NodeItem>> node_queue_
228 std::map<uint64, core::RefCountPtr<NodeItem>, std::less<uint64>>
|
| D | eager_executor.cc | 129 core::RefCountPtr<NodeItem> item(new NodeItem); in AddOrExecute()
214 void EagerExecutor::NodeDone(const core::RefCountPtr<NodeItem>& item, in NodeDone()
229 std::forward_list<core::RefCountPtr<NodeItem>> items_to_destroy; in NodeDone()
328 core::RefCountPtr<NodeItem> curr_item; in Run()
352 Status EagerExecutor::RunItem(core::RefCountPtr<NodeItem> item, in RunItem()
397 core::RefCountPtr<NodeItem> async_item(async_ref); in RunItem()
405 Status EagerExecutor::MoveToUnfinished(core::RefCountPtr<NodeItem> item, in MoveToUnfinished()
|