Searched refs:node_item (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/common_runtime/ |
D | propagator_debug_utils.cc | 43 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()
|
D | simple_propagator_state.h | 56 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()
|
D | propagator_state.h | 63 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);
|
D | simple_propagator_state.cc | 63 ",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()
|
D | propagator_debug_utils.h | 29 void DumpPendingNodeState(const NodeItem& node_item, const Entry* input_vector, 34 void DumpActiveNodeState(const NodeItem& node_item, const Entry* input_vector);
|
D | immutable_executor_state.h | 97 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()
|
D | propagator_state.cc | 72 ",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()
|
D | executor.cc | 1175 const NodeItem& item = *tagged_node.node_item; in ScheduleReady()
|