| /external/tensorflow/tensorflow/python/framework/ |
| D | subscribe.py | 83 control_outputs = {}
86 if control_input not in control_outputs:
87 control_outputs[control_input] = set()
88 control_outputs[control_input].add(op)
89 return control_outputs
101 control_outputs = self.calc_control_outputs(op.graph)
102 self.cache[op.graph] = control_outputs
104 control_outputs = self.cache[op.graph]
105 return control_outputs.get(op, [])
351 control_outputs = _ControlOutputCache()
[all …]
|
| D | function_def_to_graph.py | 83 func_graph.control_outputs = [
|
| D | func_graph.py | 189 self.control_outputs = []
1025 func_graph.control_outputs.extend(deps_control_manager.ops_which_must_run)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_to_functiondef.cc | 324 std::vector<const Node*> control_outputs; in GraphToFunctionDefHelper() local
340 control_outputs.push_back(node); in GraphToFunctionDefHelper()
367 control_outputs, control_output_names, in GraphToFunctionDefHelper()
381 const std::vector<const Node*>& control_outputs, in GraphToFunctionDef() argument
546 (control_outputs.size() != control_output_names.size())) { in GraphToFunctionDef()
548 "Expected number of control outputs (", control_outputs.size(), in GraphToFunctionDef()
553 for (int i = 0; i < control_outputs.size(); ++i) { in GraphToFunctionDef()
558 signature_name = control_outputs[i]->name(); in GraphToFunctionDef()
568 node_names.Lookup(control_outputs[i]->name()); in GraphToFunctionDef()
|
| D | graph_to_functiondef.h | 39 const std::vector<const Node*>& control_outputs,
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 43 std::vector<ControlOutput> control_outputs, const int graph_def_version, in GrapplerFunctionItem() argument
49 control_outputs_(std::move(control_outputs)), in GrapplerFunctionItem()
100 const std::vector<ControlOutput>& GrapplerFunctionItem::control_outputs() in control_outputs() function in tensorflow::grappler::GrapplerFunctionItem
275 std::vector<ControlOutput> control_outputs; in MakeGrapplerFunctionItem() local
276 control_outputs.reserve(func.control_ret_size()); in MakeGrapplerFunctionItem()
278 control_outputs.push_back({control_ret.first, control_ret.second}); in MakeGrapplerFunctionItem()
285 std::move(outputs), std::move(control_outputs), graph_def_version, in MakeGrapplerFunctionItem()
556 for (const ControlOutput& control_out : item.control_outputs()) { in MakeFunctionDef()
|
| D | functions.h | 75 const std::vector<ControlOutput>& control_outputs() const;
100 std::vector<ControlOutput> control_outputs,
|
| D | functions_test.cc | 503 const ControlOutput &ctrl = item.control_outputs()[0]; in TEST_F()
|
| /external/tensorflow/tensorflow/python/client/ |
| D | tf_session_helper.cc | 563 std::vector<TF_Operation*> control_outputs( in TF_OperationGetControlOutputs_wrapper() local
565 TF_OperationGetControlOutputs(oper, control_outputs.data(), in TF_OperationGetControlOutputs_wrapper()
566 control_outputs.size()); in TF_OperationGetControlOutputs_wrapper()
567 return control_outputs; in TF_OperationGetControlOutputs_wrapper()
588 const std::vector<TF_Operation*>* control_outputs, in TF_GraphToFunction_wrapper() argument
621 control_outputs == nullptr ? 0 : control_outputs->size(), in TF_GraphToFunction_wrapper()
622 control_outputs == nullptr ? nullptr : control_outputs->data(), in TF_GraphToFunction_wrapper()
|
| D | tf_session_helper.h | 211 const std::vector<TF_Operation*>* control_outputs,
|
| D | tf_session_wrapper.cc | 196 const std::vector<TF_Operation*> control_outputs, in PYBIND11_MODULE()
215 outputs, output_names_name_vector, &control_outputs, in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_xla_computations_pass.cc | 297 absl::flat_hash_set<Node*> control_outputs; in BuildXlaLaunchOps() local
303 control_outputs.insert(le->dst()); in BuildXlaLaunchOps()
319 AddControlOutputs(*output_node, &control_outputs); in BuildXlaLaunchOps()
345 control_outputs.erase(node); in BuildXlaLaunchOps()
366 for (Node* n : control_outputs) { in BuildXlaLaunchOps()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | graph-as-function-retval-of-arg.pbtxt | 41 # CHECK-SAME: control_outputs = ""
|
| D | graph-as-function-control-ret.pbtxt | 190 # CHECK-SAME: control_outputs = "var1_add,var2_add"
|
| D | graph-as-function.pbtxt | 8 # CHECK-SAME: control_outputs = ""
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | mlir_roundtrip_flags.h | 47 std::vector<string> control_outputs; member
|
| D | tf_mlir_translate.cc | 71 ParseOutputArrayInfo(control_output_arrays, &specs.control_outputs)); in GraphdefToMlirImport()
81 for (const auto& control_output : specs.control_outputs) { in GraphdefToMlirImport()
|
| D | import_model.cc | 1801 llvm::ArrayRef<std::string> control_outputs,
1838 specs.control_outputs, &control_ret_nodes)); in Convert()
1854 mlir::interleave(specs.control_outputs, ss, ","); in Convert()
1855 auto control_outputs = in Convert() local
1860 b.getDictionaryAttr({inputs, outputs, control_outputs}))); in Convert()
2075 llvm::ArrayRef<std::string> control_outputs, in GetControlRetsFromFunctionGraph() argument
2077 if (control_outputs.empty()) return Status::OK(); in GetControlRetsFromFunctionGraph()
2080 for (auto control_and_idx : llvm::enumerate(control_outputs)) in GetControlRetsFromFunctionGraph()
2083 if (controls_to_idx.size() != control_outputs.size()) in GetControlRetsFromFunctionGraph()
2093 for (auto node_and_name : llvm::zip(*control_ret_nodes, control_outputs)) in GetControlRetsFromFunctionGraph()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function.cc | 155 int ncontrol_outputs, const TF_Operation* const* control_outputs, in TF_GraphToFunctionWithControlOutputs() argument
200 control_output_nodes.push_back(&control_outputs[i]->node); in TF_GraphToFunctionWithControlOutputs()
|
| D | c_api.h | 486 TF_Operation* oper, TF_Operation** control_outputs,
1122 int ncontrol_outputs, const TF_Operation* const* control_outputs,
|
| D | c_api.cc | 1215 TF_Operation** control_outputs, in TF_OperationGetControlOutputs() argument
1221 control_outputs[count] = ToOperation(edge->dst()); in TF_OperationGetControlOutputs()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | dependency_optimizer.cc | 512 std::vector<gtl::InlinedVector<std::pair<int, int>, 2>> control_outputs( in TransitiveReduction() local
537 control_outputs[input_node_idx].emplace_back(node_idx, input_slot); in TransitiveReduction()
573 for (const auto& control_output : control_outputs[source]) { in TransitiveReduction()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | wrap_function.py | 327 pruned_graph.control_outputs.extend(
|
| D | wrap_function_test.py | 353 [operation.name for operation in f_pruned.graph.control_outputs])
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load_test.py | 185 exported_graph.control_outputs)
195 imported_graph.control_outputs)
|