| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | transform_graph_function.cc | 58 for (const auto& control_ret : fdef.control_ret()) { in TransformGraphFunction() local
59 node_name_to_control_ret.emplace(control_ret.second, control_ret.first); in TransformGraphFunction()
104 for (const auto& control_ret : control_ret_node_names) { in TransformGraphFunction() local
105 node_name_to_control_ret.emplace(control_ret, control_ret); in TransformGraphFunction()
108 for (const auto& control_ret : fdef.control_ret()) { in TransformGraphFunction() local
109 node_name_to_control_ret.emplace(control_ret.second, control_ret.first); in TransformGraphFunction()
171 const auto control_ret = in TransformGraphFunction() local
180 GraphToFunctionDef(*graph, func_name, control_ret, &new_func)); in TransformGraphFunction()
|
| /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 | 881 for (const auto& control_ret : control_ret_node_names) { in InstantiateMultiDevice() local
882 node_name_to_control_ret.emplace(control_ret, control_ret); in InstantiateMultiDevice()
885 for (const auto& control_ret : fdef->control_ret()) { in InstantiateMultiDevice() local
886 node_name_to_control_ret.emplace(control_ret.second, control_ret.first); in InstantiateMultiDevice()
1015 const auto control_ret = in InstantiateMultiDevice() local
1065 &control_ret, &options, status, &counter, &data] { in InstantiateMultiDevice()
1085 GraphToFunctionDef(*subgraph, unique_name, control_ret, &shard)); in InstantiateMultiDevice()
|
| /external/tensorflow/tensorflow/core/ir/importexport/tests/graphdef_to_mlir/ |
| D | invalid_missing_control_result.pbtxt | 3 # CHECK: Control output 'missing' was not found in 'control_ret'
39 control_ret {
|
| D | arg_as_control_ret.pbtxt | 45 control_ret {
49 control_ret {
|
| D | invalid_function_import.pbtxt | 5 # in the function's `control_ret` map.
7 # CHECK: INVALID_ARGUMENT: Control output 'const' was not found in 'control_ret'
|
| D | invalid_op_name.pbtxt | 15 control_ret {
|
| D | invalid_backedge_input_size.pbtxt | 16 control_ret {
|
| D | invalid_generic_func_with_empty_control_result.pbtxt | 21 control_ret {
|
| D | invalid_missing_data_result_value.pbtxt | 39 control_ret {
|
| D | invalid_missing_control_result_value.pbtxt | 39 control_ret {
|
| D | invalid_missing_data_result.pbtxt | 39 control_ret {
|
| /external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/ |
| D | test27.pbtxt | 192 control_ret {
196 control_ret {
|
| D | graph-function-control-ret-same-island.pbtxt | 80 control_ret {
|
| D | test11.pbtxt | 699 control_ret {
703 control_ret {
1062 control_ret {
1066 control_ret {
|
| D | graph-function-control-ret-diff-island.pbtxt | 91 control_ret {
|
| D | test18.pbtxt | 109 control_ret {
|
| /external/tensorflow/tensorflow/core/framework/ |
| 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_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,
|
| /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/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/ |
| D | function-control-ret.mlir | 18 // CHECK: control_ret {
|
| /external/tensorflow/tensorflow/core/ir/importexport/ |
| D | graphdef_import.cc | 614 for (const std::string &control_ret : signature.control_output()) { in ConvertFunctionDef() local
615 auto ret_it = function.control_ret().find(control_ret); in ConvertFunctionDef()
616 if (ret_it == function.control_ret().end()) { in ConvertFunctionDef()
617 return InvalidArgument("Control output '", control_ret, in ConvertFunctionDef()
627 dialect_->getTfgNameAttrIdentifier(), b_.getStringAttr(control_ret)))); in ConvertFunctionDef()
|
| /external/tensorflow/tensorflow/core/data/service/testdata/ |
| D | interleave_textline_dataset.pbtxt | 279 control_ret {
482 control_ret {
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | graph-function-control-ret-same-island.pbtxt | 92 control_ret {
|