Home
last modified time | relevance | path

Searched refs:tagged_node (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dsimple_propagator_state.h115 void PropagateOutputs(const TaggedNode& tagged_node, EntryVector* outputs,
120 Entry* GetInputTensors(const TaggedNode& tagged_node) { in GetInputTensors() argument
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()
132 FrameAndIter GetFrameAndIter(const TaggedNode& tagged_node) const { in GetFrameAndIter() argument
140 void MaybeMarkStarted(const TaggedNode& tagged_node) { in MaybeMarkStarted() argument
145 (*active_)[tagged_node.node_item->node_id] = true; in MaybeMarkStarted()
148 void MaybeMarkCompleted(const TaggedNode& tagged_node) { in MaybeMarkCompleted() argument
153 (*active_)[tagged_node.node_item->node_id] = false; in MaybeMarkCompleted()
Dpropagator_state.h433 void PropagateOutputs(const TaggedNode& tagged_node, EntryVector* outputs,
442 Entry* GetInputTensors(const TaggedNode& tagged_node) const in GetInputTensors() argument
444 return tagged_node.input_iter->input_tensors + in GetInputTensors()
445 tagged_node.node_item->input_start; in GetInputTensors()
448 FrameAndIter GetFrameAndIter(const TaggedNode& tagged_node) const { in GetFrameAndIter() argument
449 return {tagged_node.input_frame->frame_id, in GetFrameAndIter()
450 tagged_node.input_iter->iter_num}; in GetFrameAndIter()
457 void MaybeMarkStarted(const TaggedNode& tagged_node) { in MaybeMarkStarted() argument
461 mutex_lock l(tagged_node.input_frame->mu); in MaybeMarkStarted()
462 tagged_node.input_iter->mark_started( in MaybeMarkStarted()
[all …]
Dexecutor.cc295 const TaggedNode& tagged_node, Entry* first_input,
498 tagged_node(_tagged_node), in AsyncState()
512 TaggedNode tagged_node; member
594 const TaggedNode& tagged_node, Entry* first_input, in ProcessAsync() argument
599 new AsyncState(params, tagged_node, &item, first_input, stats); in ProcessAsync()
613 << (state->tagged_node.get_is_dead() ? " is dead" : "") in ProcessAsync()
622 propagator_.MaybeMarkCompleted(state->tagged_node); in ProcessAsync()
625 propagator_.PropagateOutputs(state->tagged_node, &outputs, &ready); in ProcessAsync()
664 void ExecutorState<PropagatorStateType>::Process(TaggedNode tagged_node, in Process() argument
676 {{"id", step_id_}, {"iter_num", tagged_node.get_iter_num()}}); in Process()
[all …]
Dsimple_propagator_state.cc56 void SimplePropagatorState::PropagateOutputs(const TaggedNode& tagged_node, in PropagateOutputs() argument
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()
Dpropagator_state.cc65 void PropagatorState::PropagateOutputs(const TaggedNode& tagged_node, in PropagateOutputs() argument
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()
80 FrameState* const input_frame = tagged_node.input_frame; in PropagateOutputs()
81 IterationState* const input_iter = tagged_node.input_iter; in PropagateOutputs()
82 const bool is_dead = tagged_node.is_dead; in PropagateOutputs()