Home
last modified time | relevance | path

Searched refs:FunctionalizeControlFlow (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow.cc49 Status FunctionalizeControlFlow(const FunctionLibraryDefinition* lookup_library, in FunctionalizeControlFlow() function
71 Status FunctionalizeControlFlow(Graph* graph, in FunctionalizeControlFlow() function
73 return FunctionalizeControlFlow(/*lookup_library=*/nullptr, graph, library); in FunctionalizeControlFlow()
89 TF_RETURN_IF_ERROR(FunctionalizeControlFlow(lookup_library, &graph, library)); in FunctionalizeControlFlowForGraphDef()
205 TF_RETURN_IF_ERROR(FunctionalizeControlFlow(g, fld)); in FunctionalizeControlFlowForFunction()
Dfunctionalize_control_flow.h30 Status FunctionalizeControlFlow(Graph* graph,
32 Status FunctionalizeControlFlow(const FunctionLibraryDefinition* lookup_library,
Dfunctionalize_control_flow_test.cc65 TEST(FunctionalizeControlFlow, Conditional) { in TEST() argument
102 TF_ASSERT_OK(FunctionalizeControlFlow(&graph, &library)); in TEST()
202 TEST(FunctionalizeControlFlow, OneLoopVar) { in TEST() argument
256 TF_ASSERT_OK(FunctionalizeControlFlow(&graph, &library)); in TEST()
356 TEST(FunctionalizeControlFlow, NoinlineLoopBody) { in TEST() argument
408 TF_ASSERT_OK(FunctionalizeControlFlow(&lookup_lib, &graph, &library)); in TEST()
462 TEST(FunctionalizeControlFlow, MissingFunctionDefInLibrary) { in TEST() argument
489 TEST(FunctionalizeControlFlow, OneLoopVarWithoutExit) { in TEST() argument
528 TF_ASSERT_OK(FunctionalizeControlFlow(&graph, &library)); in TEST()
599 TEST(FunctionalizeControlFlow, TwoLoopVars) { in TEST() argument
[all …]
Dtf2xla.cc370 TF_RETURN_IF_ERROR(FunctionalizeControlFlow(g.get(), &flib_def)); in InitGraph()