Home
last modified time | relevance | path

Searched refs:node_item (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dpropagator_debug_utils.cc43 void DumpPendingNodeState(const NodeItem& node_item, const Entry* input_vector, in DumpPendingNodeState() argument
45 const int input_base = node_item.input_start; in DumpPendingNodeState()
48 for (int i = 0; i < node_item.num_inputs; ++i) { in DumpPendingNodeState()
60 LOG(WARNING) << " Pending Node: " << node_item.DebugString(); in DumpPendingNodeState()
61 for (int i = 0; i < node_item.num_inputs; ++i) { in DumpPendingNodeState()
75 void DumpActiveNodeState(const NodeItem& node_item, const Entry* input_vector) { in DumpActiveNodeState() argument
76 LOG(WARNING) << " Active Node: " << node_item.DebugString(); in DumpActiveNodeState()
77 const int input_base = node_item.input_start; in DumpActiveNodeState()
78 for (int i = 0; i < node_item.num_inputs; ++i) { in DumpActiveNodeState()
Dsimple_propagator_state.h56 const NodeItem* node_item; member
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()
127 CHECK_EQ(pending_[tagged_node.node_item->node_id], 0); in GetInputTensors()
129 return input_tensors_.data() + tagged_node.node_item->input_start; in GetInputTensors()
145 (*active_)[tagged_node.node_item->node_id] = true; in MaybeMarkStarted()
153 (*active_)[tagged_node.node_item->node_id] = false; in MaybeMarkCompleted()
Dpropagator_state.h63 const NodeItem* node_item; member
69 TaggedNode(const NodeItem* node_item, FrameState* in_frame, in TaggedNode()
71 : node_item(node_item), in TaggedNode()
76 const NodeItem& get_node_item() const { return *node_item; } in get_node_item()
445 tagged_node.node_item->input_start; in GetInputTensors()
463 immutable_state_.pending_ids()[tagged_node.node_item->node_id]); in MaybeMarkStarted()
473 immutable_state_.pending_ids()[tagged_node.node_item->node_id]); in MaybeMarkCompleted()
481 const NodeItem& node_item, FrameState** child);
Dsimple_propagator_state.cc63 ",kernel_name=", tagged_node.node_item->kernel->name_view(), in PropagateOutputs()
64 ",num_output_edges=", tagged_node.node_item->num_output_edges, in PropagateOutputs()
66 tagged_node.node_item->num_output_control_edges, "#"); in PropagateOutputs()
75 const NodeItem* item = tagged_node.node_item; in PropagateOutputs()
Dpropagator_debug_utils.h29 void DumpPendingNodeState(const NodeItem& node_item, const Entry* input_vector,
34 void DumpActiveNodeState(const NodeItem& node_item, const Entry* input_vector);
Dimmutable_executor_state.h97 const FrameInfo& get_enter_frame_info(const NodeItem& node_item) const { in get_enter_frame_info() argument
98 DCHECK(node_item.is_enter); in get_enter_frame_info()
99 return *enter_frame_info_[node_item.node_id]; in get_enter_frame_info()
Dpropagator_state.cc72 ",kernel_name=", tagged_node.node_item->kernel->name_view(), in PropagateOutputs()
73 ",num_output_edges=", tagged_node.node_item->num_output_edges, in PropagateOutputs()
75 tagged_node.node_item->num_output_control_edges, "#"); in PropagateOutputs()
79 const NodeItem* const item = tagged_node.node_item; in PropagateOutputs()
231 const NodeItem& node_item, in FindOrCreateChildFrame() argument
235 immutable_state_.get_enter_frame_info(node_item); in FindOrCreateChildFrame()
405 t.node_item = dst_item; \ in ActivateNodesFastPathInternal()
Dexecutor.cc1175 const NodeItem& item = *tagged_node.node_item; in ScheduleReady()