| /external/tensorflow/tensorflow/python/framework/ |
| D | subscribe.py | 84 control_outputs = {}
87 if control_input not in control_outputs:
88 control_outputs[control_input] = set()
89 control_outputs[control_input].add(op)
90 return control_outputs
102 control_outputs = self.calc_control_outputs(op.graph)
103 self.cache[op.graph] = control_outputs
105 control_outputs = self.cache[op.graph]
106 return control_outputs.get(op, [])
352 control_outputs = _ControlOutputCache()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | add.pbtxt | 45 # CHECK-SAME: control_outputs = ""
53 # SOME-SAME: control_outputs = ""
61 # NONE-SAME: control_outputs = ""
69 # UNKNOWN-SAME: control_outputs = ""
77 # DYNAMIC-SAME: control_outputs = ""
|
| D | target.pbtxt | 170 # CHECK-SAME: control_outputs = "AssignAdd"
181 # PRUNE-SAME: control_outputs = "AssignAdd"
196 # PRESERVE-SAME: control_outputs = "Variable/Assign,AssignAdd"
|
| D | multi-output-feeds.pbtxt | 273 # CHECK-SAME: control_outputs = ""
286 # PRUNE-SAME: control_outputs = ""
303 # PRESERVE-SAME: control_outputs = ""
|
| D | feed-as-fetch.pbtxt | 19 # CHECK-SAME: control_outputs = ""
|
| D | arg-as-fetch.pbtxt | 25 # CHECK-SAME: control_outputs = ""
|
| D | graph-as-function-retval-of-arg.pbtxt | 41 # CHECK-SAME: control_outputs = ""
|
| D | feed-control-dep.pbtxt | 64 # CHECK-SAME: control_outputs = ""
|
| D | graph-scalar-input.pbtxt | 7 # CHECK-SAME: control_outputs = ""
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_to_functiondef.cc | 318 std::vector<const Node*> control_outputs; in GraphToFunctionDefHelper() local
334 control_outputs.push_back(node); in GraphToFunctionDefHelper()
361 control_outputs, control_output_names, in GraphToFunctionDefHelper()
375 const std::vector<const Node*>& control_outputs, in GraphToFunctionDef() argument
550 (control_outputs.size() != control_output_names.size())) { in GraphToFunctionDef()
552 "Expected number of control outputs (", control_outputs.size(), in GraphToFunctionDef()
557 for (int i = 0; i < control_outputs.size(); ++i) { in GraphToFunctionDef()
562 signature_name = control_outputs[i]->name(); in GraphToFunctionDef()
572 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/python/client/ |
| D | tf_session_helper.cc | 558 std::vector<TF_Operation*> control_outputs( in TF_OperationGetControlOutputs_wrapper() local
560 TF_OperationGetControlOutputs(oper, control_outputs.data(), in TF_OperationGetControlOutputs_wrapper()
561 control_outputs.size()); in TF_OperationGetControlOutputs_wrapper()
562 return control_outputs; in TF_OperationGetControlOutputs_wrapper()
583 const std::vector<TF_Operation*>* control_outputs, in TF_GraphToFunction_wrapper() argument
616 control_outputs == nullptr ? 0 : control_outputs->size(), in TF_GraphToFunction_wrapper()
617 control_outputs == nullptr ? nullptr : control_outputs->data(), in TF_GraphToFunction_wrapper()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 44 std::vector<ControlOutput> control_outputs, const int graph_def_version, in GrapplerFunctionItem() argument
51 control_outputs_(std::move(control_outputs)), in GrapplerFunctionItem()
102 const std::vector<ControlOutput>& GrapplerFunctionItem::control_outputs() in control_outputs() function in tensorflow::grappler::GrapplerFunctionItem
282 std::vector<ControlOutput> control_outputs; in MakeGrapplerFunctionItem() local
283 control_outputs.reserve(func.control_ret_size()); in MakeGrapplerFunctionItem()
285 control_outputs.push_back({control_ret.first, control_ret.second}); in MakeGrapplerFunctionItem()
297 std::move(inputs), std::move(outputs), std::move(control_outputs), in MakeGrapplerFunctionItem()
571 for (const ControlOutput& control_out : item.control_outputs()) { in MakeFunctionDef()
|
| D | functions.h | 75 const std::vector<ControlOutput>& control_outputs() const;
102 std::vector<ControlOutput> control_outputs,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tf_saved_model_lift_variables.mlir | 8 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
32 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
42 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
70 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
|
| D | tf_saved_model_lift_variables_invalid_session.mlir | 8 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
25 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
|
| D | tf_data_fuse_map_and_batch.mlir | 5 func @fuse_map_and_batch() -> tensor<!tf.variant> attributes {tf.entry_function = {control_outputs …
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_xla_computations_pass.cc | 291 absl::flat_hash_set<Node*> control_outputs; in BuildXlaLaunchOps() local
297 control_outputs.insert(le->dst()); in BuildXlaLaunchOps()
313 AddControlOutputs(*output_node, &control_outputs); in BuildXlaLaunchOps()
339 control_outputs.erase(node); in BuildXlaLaunchOps()
360 for (Node* n : control_outputs) { in BuildXlaLaunchOps()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | insert_call_once_op.mlir | 19 …attributes {tf.entry_function = {control_outputs = "", inputs = "input:0", outputs = "hash_table_L…
35 …attributes {tf.entry_function = {control_outputs = "", inputs = "input:0", outputs = "output:0"}, …
|
| D | initialize_variables.mlir | 7 …d_model.index_path = ["output_0"]}) attributes {tf.entry_function = {control_outputs = "", inputs …
48 …d_model.index_path = ["output_0"]}) attributes {tf.entry_function = {control_outputs = "", inputs …
|
| D | legalize-tf-variables.mlir | 7 …d_model.index_path = ["output_0"]}) attributes {tf.entry_function = {control_outputs = "", inputs …
42 …d_model.index_path = ["output_0"]}) attributes {tf.entry_function = {control_outputs = "", inputs …
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/end2end/ |
| D | graph-input-node.pbtxt | 52 # CHECK-SAME: control_outputs = ""
|
| D | custom_opdef.pbtxt | 40 # CHECK: attributes {tf.entry_function = {control_outputs = "", inputs = "input0,input1", outputs =…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | mlir_roundtrip_flags.h | 46 std::vector<string> control_outputs; member
|
| D | import_model.cc | 698 for (const auto& control_output : specs_.control_outputs) in GetInputOutputNodes()
2202 llvm::ArrayRef<std::string> control_outputs,
2237 TF_RETURN_IF_ERROR(importer.GetControlRetsFromGraph(specs.control_outputs, in Convert()
2252 llvm::interleave(specs.control_outputs, ss, ","); in Convert()
2253 auto control_outputs = in Convert() local
2260 b.getDictionaryAttr({inputs, outputs, control_outputs}))); in Convert()
2267 TF_RETURN_IF_ERROR(importer.GetControlRetsFromGraph(specs.control_outputs, in Convert()
2274 !specs.control_outputs.empty()) { in Convert()
2287 llvm::interleave(specs.control_outputs, ss, ","); in Convert()
2288 auto control_outputs = in Convert() local
[all …]
|