Searched refs:FunctionalizeControlFlow (Results 1 - 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | functionalize_control_flow.h | 26 Status FunctionalizeControlFlow(Graph* graph,
|
| H A D | functionalize_control_flow_test.cc | 63 TEST(FunctionalizeControlFlow, Conditional) {
96 TF_ASSERT_OK(FunctionalizeControlFlow(&graph, &library));
189 TEST(FunctionalizeControlFlow, OneLoopVar) {
239 TF_ASSERT_OK(FunctionalizeControlFlow(&graph, &library));
307 TEST(FunctionalizeControlFlow, OneLoopVarWithoutExit) {
342 TF_ASSERT_OK(FunctionalizeControlFlow(&graph, &library));
410 TEST(FunctionalizeControlFlow, TwoLoopVars) {
483 TF_ASSERT_OK(FunctionalizeControlFlow(&graph, &library));
584 TEST(FunctionalizeControlFlow, Complex) {
716 TF_ASSERT_OK(FunctionalizeControlFlow(
[all...] |
| H A D | xla_compiler.cc | 604 FunctionalizeControlFlow(graph.get(), local_flib_def_.get()));
|
| H A D | functionalize_control_flow.cc | 919 LOG(INFO) << "FunctionalizeControlFlow (with_clusters): "
1095 LOG(INFO) << "FunctionalizeControlFlow (before XlaIf conversion): "
1138 LOG(INFO) << "FunctionalizeControlFlow (after XlaIf conversion): "
1343 Status FunctionalizeControlFlow(Graph* graph, function
1345 VLOG(2) << "FunctionalizeControlFlow (initial): "
1422 // FunctionalizeControlFlow is invoked for every function, so the loops's
1427 VLOG(2) << "FunctionalizeControlFlow (final): "
|
Completed in 107 milliseconds