Searched refs:pending_inputs (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | static_schedule.cc | 63 std::unordered_map<const NodeDef*, int> pending_inputs; in EstimateEarliestExecutionTimes() local
73 pending_inputs[&node] = 1; in EstimateEarliestExecutionTimes()
75 pending_inputs[&node] = node.input_size(); in EstimateEarliestExecutionTimes()
110 int pending = pending_inputs[fanout]; in EstimateEarliestExecutionTimes()
117 pending_inputs[fanout]--; in EstimateEarliestExecutionTimes()
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | debug_data.py | 846 pending_inputs = {}
848 pending_inputs[node] = []
860 (inp_node, inp_output_slot) not in pending_inputs[node]):
861 pending_inputs[node].append((inp_node, inp_output_slot))
871 device_name, pending_inputs[node], datum.timestamp, start_i=i + 1):
875 (node, datum.timestamp, repr(pending_inputs[node])))
879 recipient_pending_inputs = pending_inputs[recipient]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session.cc | 940 auto it = run_state->pending_inputs.find(input.first); in PRun()
941 if (it == run_state->pending_inputs.end()) { in PRun()
994 auto it = run_state->pending_inputs.find(input.first); in PRun()
1135 for (const auto& input : run_state->pending_inputs) { in CheckFetch()
1656 pending_inputs[name] = false;
1678 for (const auto& it : pending_inputs) { in PendingDone()
|
| D | direct_session.h | 187 std::unordered_map<string, bool> pending_inputs; // true if fed member
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_session.h | 176 std::unordered_map<string, bool> pending_inputs; // true if fed member
|
| D | master_session.cc | 984 for (const auto& input : run_state->pending_inputs) { in CheckFetches()
1681 auto it = run_state->pending_inputs.find(feed); in DoPartialRun()
1682 if (it == run_state->pending_inputs.end()) { in DoPartialRun()
1712 auto it = run_state->pending_inputs.find(req.feed_name(i)); in DoPartialRun()
2052 pending_inputs[name] = false; in RunState()
2064 for (const auto& it : pending_inputs) { in PendingDone()
|