Searched refs:cond_fdef (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util_test.cc | 349 FunctionDef cond_fdef; in TEST() local
350 TF_ASSERT_OK(GraphToFunctionDef(graph, "cond", &cond_fdef)); in TEST()
351 TF_ASSERT_OK(fld.AddFunctionDef(cond_fdef)); in TEST()
382 FunctionDef cond_fdef; in TEST() local
383 TF_ASSERT_OK(GraphToFunctionDef(graph, "cond", &cond_fdef)); in TEST()
384 TF_ASSERT_OK(fld.AddFunctionDef(cond_fdef)); in TEST()
|
| D | functionalize_while.cc | 423 FunctionDef cond_fdef; in FunctionalizeLoop() local
425 GraphToFunctionDef(*cond_graph, cond_name.name(), &cond_fdef)); in FunctionalizeLoop()
430 TF_RETURN_IF_ERROR(library->AddFunctionDef(cond_fdef)); in FunctionalizeLoop()
|
| D | functionalize_control_flow_test.cc | 223 FunctionDef cond_fdef; in RunTest() local
224 TF_ASSERT_OK(GraphToFunctionDef(cond_graph, "cond_fn", &cond_fdef)); in RunTest()
227 *(fdef_lib.add_function()) = cond_fdef; in RunTest()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1778 const FunctionDef* cond_fdef = fld->Find(cond_func.name()); in FindArgsToLiftForWhileNode() local
1779 if (cond_fdef == nullptr) { in FindArgsToLiftForWhileNode()
1785 *cond_fdef, AttrSlice(&cond_func.attr()), fld, &cond_fbody)); in FindArgsToLiftForWhileNode()
|