| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_body.cc | 51 for (const auto& control_ret : fdef.control_ret()) { in FunctionBody() local
52 control_ret_node_names.insert(control_ret.second); in FunctionBody()
|
| D | process_function_library_runtime.cc | 776 for (const auto& control_ret : control_ret_node_names) { in InstantiateMultiDevice() local
777 node_name_to_control_ret.emplace(control_ret, control_ret); in InstantiateMultiDevice()
780 for (const auto& control_ret : fdef->control_ret()) { in InstantiateMultiDevice() local
781 node_name_to_control_ret.emplace(control_ret.second, control_ret.first); in InstantiateMultiDevice()
900 const auto control_ret = in InstantiateMultiDevice() local
932 &control_ret, &options, status, &counter, &data] { in InstantiateMultiDevice()
949 GraphToFunctionDef(*subgraph, unique_name, control_ret, &shard)); in InstantiateMultiDevice()
|
| D | function.cc | 880 for (const auto& control_ret : fdef.control_ret()) { in PruneFunctionBody() local
881 control_ret_nodes.insert(control_ret.second); in PruneFunctionBody()
|
| /external/tensorflow/tensorflow/core/framework/ |
| 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.cc | 297 const std::function<absl::optional<string>(const Node*)>& control_ret, in GraphToFunctionDefHelper() argument
331 if (control_ret) { in GraphToFunctionDefHelper()
332 auto control_ret_name = control_ret(node); in GraphToFunctionDefHelper()
588 const std::function<absl::optional<string>(const Node*)>& control_ret, in GraphToFunctionDef() argument
590 return GraphToFunctionDefHelper(graph, name, control_ret, in GraphToFunctionDef()
|
| 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 | 102 map<string, string> control_ret = 6; field
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/ |
| D | function-control-ret.mlir | 18 // CHECK: control_ret {
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 284 for (const auto& control_ret : func.control_ret()) { in MakeGrapplerFunctionItem() local
285 control_outputs.push_back({control_ret.first, control_ret.second}); in MakeGrapplerFunctionItem()
|
| /external/llvm-project/llvm/unittests/Analysis/Inputs/ir2native_x86_64_model/ |
| D | saved_model.pbtxt | 1723 control_ret {
1727 control_ret {
1731 control_ret {
1735 control_ret {
1739 control_ret {
1743 control_ret {
1747 control_ret {
1751 control_ret {
1755 control_ret {
1759 control_ret {
[all …]
|
| /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 {
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function_def_to_graph.py | 89 func_graph.get_operation_by_name(fdef.control_ret[ret_name])
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 629 const auto control_ret = [&](const Node* n) -> absl::optional<string> { in ConvertLibFunction() local
636 GraphToFunctionDef(*sub_graph, function_name, control_ret, &func_def)); in ConvertLibFunction()
|
| D | import_model.cc | 1582 for (Node* control_ret : control_ret_nodes) { in ConvertFunctionArgAndRets()
1583 auto* inst = node_values_[control_ret->id()]; in ConvertFunctionArgAndRets()
|
| /external/llvm-project/llvm/lib/Analysis/models/inliner/ |
| D | saved_model.pbtxt | 1629 control_ret {
19400 control_ret {
19404 control_ret {
19408 control_ret {
19917 control_ret {
20686 control_ret {
21455 control_ret {
22003 control_ret {
22357 control_ret {
23126 control_ret {
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/tests/ |
| D | opens2s_gnmt_mixed_precision.pbtxt.gz | |