| /external/tensorflow/tensorflow/contrib/graph_editor/ |
| D | select.py | 205 def check_cios(control_inputs=False, control_outputs=None, control_ios=None): argument
229 if control_outputs is not None:
232 control_outputs = control_ios
233 elif control_outputs is not None:
234 if not isinstance(control_outputs, util.ControlOutputs):
236 type(control_outputs)))
238 if control_outputs is not None:
239 control_outputs.update()
240 return control_inputs, control_outputs
243 def get_ops_ios(ops, control_inputs=False, control_outputs=None, argument
[all …]
|
| D | edit.py | 51 def detach_control_outputs(sgv, control_outputs): argument
59 if not isinstance(control_outputs, util.ControlOutputs):
61 type(control_outputs))
62 control_outputs.update()
65 for cop in control_outputs.get(op):
101 def detach_outputs(sgv, control_outputs=None): argument
136 if control_outputs is not None:
137 detach_control_outputs(sgv_, control_outputs)
141 def detach(sgv, control_inputs=False, control_outputs=None, control_ios=None): argument
164 control_inputs, control_outputs = select.check_cios(control_inputs,
[all …]
|
| D | transform.py | 332 self.control_outputs = util.ControlOutputs(sgv.graph)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | subscribe.py | 81 control_outputs = {}
84 if control_input not in control_outputs:
85 control_outputs[control_input] = set()
86 control_outputs[control_input].add(op)
87 return control_outputs
98 control_outputs = self.calc_control_outputs(op.graph)
99 self.cache[op.graph] = control_outputs
101 control_outputs = self.cache[op.graph]
102 return control_outputs.get(op, [])
342 control_outputs = _ControlOutputCache()
[all …]
|
| D | function_def_to_graph.py | 76 func_graph.control_outputs = [
|
| D | func_graph.py | 181 self.control_outputs = []
734 func_graph.control_outputs.extend(control_manager.ops_which_must_run)
|
| /external/tensorflow/tensorflow/contrib/graph_editor/tests/ |
| D | util_test.py | 116 control_outputs = ge.util.ControlOutputs(g0).get_all()
117 self.assertEqual(len(control_outputs), 1)
118 self.assertEqual(len(control_outputs[x0.op]), 1)
119 self.assertIs(list(control_outputs[x0.op])[0], c0.op)
|
| D | edit_test.py | 54 control_outputs = ge.ControlOutputs(self.graph)
55 ge.detach(sgv, control_ios=control_outputs)
|
| D | select_test.py | 78 control_outputs = ge.util.ControlOutputs(self.graph)
80 len(ge.get_ops_ios(self.h.op, control_ios=control_outputs)), 3)
83 len(ge.get_ops_ios(self.c.op, control_ios=control_outputs)), 6)
116 control_outputs = ge.util.ControlOutputs(self.graph)
122 control_ios=control_outputs)
|
| /external/tensorflow/tensorflow/python/client/ |
| D | tf_session_helper.cc | 569 std::vector<TF_Operation*> control_outputs( in TF_OperationGetControlOutputs_wrapper() local
571 TF_OperationGetControlOutputs(oper, control_outputs.data(), in TF_OperationGetControlOutputs_wrapper()
572 control_outputs.size()); in TF_OperationGetControlOutputs_wrapper()
573 return control_outputs; in TF_OperationGetControlOutputs_wrapper()
594 const std::vector<TF_Operation*>* control_outputs, in TF_GraphToFunction_wrapper() argument
627 control_outputs == nullptr ? 0 : control_outputs->size(), in TF_GraphToFunction_wrapper()
628 control_outputs == nullptr ? nullptr : control_outputs->data(), in TF_GraphToFunction_wrapper()
|
| D | tf_session.i | 608 %typemap(in) (const std::vector<TF_Operation*>* control_outputs)
609 (std::vector<TF_Operation*> control_outputs) {
620 control_outputs.push_back(oper_ptr);
622 $1 = &control_outputs;
|
| D | tf_session_helper.h | 212 const std::vector<TF_Operation*>* control_outputs,
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function.cc | 305 const std::vector<const Node*>& control_outputs, in GraphToFunctionDef() argument
431 (control_outputs.size() != control_output_names.size())) { in GraphToFunctionDef()
433 "Expected number of control outputs (", control_outputs.size(), in GraphToFunctionDef()
438 for (int i = 0; i < control_outputs.size(); ++i) { in GraphToFunctionDef()
443 signature_name = control_outputs[i]->name(); in GraphToFunctionDef()
450 (*fdef->mutable_control_ret())[signature_name] = control_outputs[i]->name(); in GraphToFunctionDef()
565 int ncontrol_outputs, const TF_Operation* const* control_outputs, in TF_GraphToFunctionWithControlOutputs() argument
610 control_output_nodes.push_back(&control_outputs[i]->node); in TF_GraphToFunctionWithControlOutputs()
|
| D | c_api.h | 686 TF_Operation* oper, TF_Operation** control_outputs,
1335 int ncontrol_outputs, const TF_Operation* const* control_outputs,
|
| D | c_api.cc | 1628 TF_Operation** control_outputs, in TF_OperationGetControlOutputs() argument
1634 control_outputs[count] = ToOperation(edge->dst()); in TF_OperationGetControlOutputs()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_xla_computations_pass.cc | 296 absl::flat_hash_set<Node*> control_outputs; in BuildXlaLaunchOps() local
302 control_outputs.insert(le->dst()); in BuildXlaLaunchOps()
318 AddControlOutputs(*output_node, &control_outputs); in BuildXlaLaunchOps()
344 control_outputs.erase(node); in BuildXlaLaunchOps()
365 for (Node* n : control_outputs) { in BuildXlaLaunchOps()
|
| D | encapsulate_subgraphs_pass.cc | 403 std::unordered_set<const Node*> control_outputs; member
863 outside_subgraph->control_outputs.insert(edge->dst()); in RecordOutsideCompilationOutputOrControl()
946 !oc_subgraph.control_outputs.empty()) { in AddHostComputes()
1035 for (const auto& dst_node : oc_subgraph.control_outputs) { in AddHostComputes()
1367 !oc_subgraph.control_outputs.empty()) { in AddOutsideCompilationHostIONodes()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 314 std::vector<ControlOutput> control_outputs, const int graph_def_version, in GrapplerFunctionItem() argument
320 control_outputs_(std::move(control_outputs)), in GrapplerFunctionItem()
374 const std::vector<ControlOutput>& GrapplerFunctionItem::control_outputs() in control_outputs() function in tensorflow::grappler::GrapplerFunctionItem
641 std::vector<ControlOutput> control_outputs; in MakeGrapplerFunctionItem() local
642 control_outputs.reserve(func.control_ret_size()); in MakeGrapplerFunctionItem()
644 control_outputs.push_back({control_ret.first, control_ret.second}); in MakeGrapplerFunctionItem()
651 std::move(outputs), std::move(control_outputs), graph_def_version, in MakeGrapplerFunctionItem()
865 for (const ControlOutput& control_out : item.control_outputs()) { in MakeFunctionDef()
|
| D | functions.h | 170 const std::vector<ControlOutput>& control_outputs() const;
195 std::vector<ControlOutput> control_outputs,
|
| D | functions_test.cc | 676 const ControlOutput &ctrl = item.control_outputs()[0]; in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | dependency_optimizer.cc | 480 std::vector<gtl::InlinedVector<std::pair<int, int>, 2>> control_outputs( in TransitiveReduction() local
500 control_outputs[input_node_idx].emplace_back(node_idx, input_slot); in TransitiveReduction()
520 for (const auto& control_output : control_outputs[source]) { in TransitiveReduction()
551 for (const auto& control_output : control_outputs[source]) { in TransitiveReduction()
|
| D | function_optimizer.cc | 1836 for (const ControlOutput& control_output : item.control_outputs()) { in InlineIndirectFunctionCall()
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load_test.py | 110 exported_graph.control_outputs)
120 imported_graph.control_outputs)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function.py | 326 [o._c_op for o in graph.control_outputs], # pylint: disable=protected-access
|