| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_to_functiondef.cc | 303 const std::function<absl::optional<string>(const Node*)>& control_ret, in GraphToFunctionDefHelper() argument
337 if (control_ret) { in GraphToFunctionDefHelper()
338 auto control_ret_name = control_ret(node); in GraphToFunctionDefHelper()
584 const std::function<absl::optional<string>(const Node*)>& control_ret, in GraphToFunctionDef() argument
586 return GraphToFunctionDefHelper(graph, name, control_ret, in GraphToFunctionDef()
|
| D | graph_to_functiondef_test.cc | 143 const auto control_ret = [](const Node* n) -> absl::optional<string> { in TEST() local
149 TF_EXPECT_OK(GraphToFunctionDef(*graph, "test_fn", control_ret, &fdef)); in TEST()
|
| D | graph_to_functiondef.h | 53 const std::function<absl::optional<string>(const Node*)>& control_ret,
|
| D | function.cc | 599 for (const auto& cr : fdef.control_ret()) { in Print()
880 std::map<string, string> control_ret1(f1.control_ret().begin(), in FunctionDefsEqual()
881 f1.control_ret().end()); in FunctionDefsEqual()
882 std::map<string, string> control_ret2(f2.control_ret().begin(), in FunctionDefsEqual()
883 f2.control_ret().end()); in FunctionDefsEqual()
911 std::map<string, string> control_ret(fdef.control_ret().begin(), in FunctionDefHash() local
912 fdef.control_ret().end()); in FunctionDefHash()
913 for (const auto& p : control_ret) { in FunctionDefHash()
|
| D | function.proto | 100 map<string, string> control_ret = 6; field
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 277 for (const auto& control_ret : func.control_ret()) { in MakeGrapplerFunctionItem() local
278 control_outputs.push_back({control_ret.first, control_ret.second}); in MakeGrapplerFunctionItem()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/ |
| D | function-control-ret.mlir | 18 // CHECK: control_ret {
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | graph-function-control-ret-same-island.pbtxt | 92 control_ret {
|
| D | graph-function-control-ret-diff-island.pbtxt | 103 control_ret {
|
| D | graph-function-variable-shapes.pbtxt | 163 control_ret {
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 773 for (const auto& control_ret : fdef->control_ret()) { in InstantiateMultiDevice() local
774 node_name_to_control_ret.emplace(control_ret.second, control_ret.first); in InstantiateMultiDevice()
779 const auto control_ret = in InstantiateMultiDevice() local
810 runner([this, &pair, comp_data, unique_name, data_lib_def, &control_ret, in InstantiateMultiDevice()
828 GraphToFunctionDef(*subgraph, unique_name, control_ret, &shard)); in InstantiateMultiDevice()
|
| D | function.cc | 900 for (const auto& control_ret : fdef.control_ret()) { in PruneFunctionBody() local
901 control_ret_nodes.insert(control_ret.second); in PruneFunctionBody()
2370 for (const auto& control_ret : fdef.control_ret()) { in FunctionBody() local
2371 control_ret_node_names.insert(control_ret.second); in FunctionBody()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | convert_to_constants.py | 597 for key, value in input_library_func.control_ret.items():
598 output_library_func.control_ret[key] = value
|
| D | function_def_to_graph.py | 84 func_graph.get_operation_by_name(fdef.control_ret[ret_name])
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 732 const auto control_ret = [&](const Node* n) -> absl::optional<string> { in ConvertLibFunction() local
739 GraphToFunctionDef(*sub_graph, function_name, control_ret, &func_def)); in ConvertLibFunction()
|
| D | import_model.cc | 1262 for (Node* control_ret : control_ret_nodes) { in ConvertFunctionArgAndRets()
1263 auto* inst = node_values_[control_ret->id()]; in ConvertFunctionArgAndRets()
|
| /external/tensorflow/tensorflow/compiler/jit/tests/ |
| D | opens2s_gnmt_mixed_precision.pbtxt.gz | |