Home
last modified time | relevance | path

Searched refs:ComputeWhileLoopTripCount (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_transformer_test.cc131 auto result = ComputeWhileLoopTripCount(while_hlo); in TEST_F()
142 auto result = ComputeWhileLoopTripCount(while_hlo); in TEST_F()
153 auto result = ComputeWhileLoopTripCount(while_hlo); in TEST_F()
164 auto result = ComputeWhileLoopTripCount(while_hlo); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_analysis.h31 absl::optional<int64> ComputeWhileLoopTripCount(
Dwhile_loop_trip_count_annotator.cc29 if (auto trip_count = ComputeWhileLoopTripCount(instr)) { in Run()
Dwhile_loop_analysis.cc471 optional<int64> ComputeWhileLoopTripCount(HloInstruction* while_op, in ComputeWhileLoopTripCount() function
562 auto exact_trip_count = ComputeWhileLoopTripCount(while_op); in ComputeWhileLoopTripCountUpperBound()
Dwhile_loop_simplifier.cc660 ComputeWhileLoopTripCount(while_op, /*max_brute_force_iters=*/1); in TryRemoveWhileLoop()