Searched refs:FunctionalizeControlFlow (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.cc | 248 TF_RETURN_IF_ERROR(FunctionalizeControlFlow(g, fld, node_filter)); in FunctionalizeControlFlowForFunction()
273 Status FunctionalizeControlFlow(Graph* graph, in FunctionalizeControlFlow() function
319 TF_RETURN_IF_ERROR(FunctionalizeControlFlow(&graph, library, node_filter, in FunctionalizeControlFlowForGraphDef()
397 return FunctionalizeControlFlow(options.graph->get(), options.flib_def); in Run()
|
| D | functionalize_control_flow_test.cc | 91 FunctionalizeControlFlow, ConditionalTestFixture,
254 TF_ASSERT_OK(FunctionalizeControlFlow( in RunTest()
310 TEST(FunctionalizeControlFlow, OneLoopVar) { in TEST() argument
364 TF_ASSERT_OK(FunctionalizeControlFlow(&graph, &library)); in TEST()
464 TEST(FunctionalizeControlFlow, NoinlineLoopBody) { in TEST() argument
514 TF_ASSERT_OK(FunctionalizeControlFlow(&graph, &library)); in TEST()
568 TEST(FunctionalizeControlFlow, MissingFunctionDefInLibrary) { in TEST() argument
593 TEST(FunctionalizeControlFlow, OneLoopVarWithoutExit) { in TEST() argument
632 TF_ASSERT_OK(FunctionalizeControlFlow(&graph, &library)); in TEST()
703 TEST(FunctionalizeControlFlow, TwoLoopVars) { in TEST() argument
[all …]
|
| D | functionalize_control_flow.h | 51 Status FunctionalizeControlFlow(Graph* graph,
|
| D | graph_compiler_util.cc | 303 TF_RETURN_IF_ERROR(FunctionalizeControlFlow(g.get(), &flib_def)); in InitGraph()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | upgrade_graph.cc | 189 FunctionalizeControlFlow(graph, flib_def, node_filter, in UpgradeLegacyGraph()
|