HomeSort by relevance Sort by last modified time
    Searched refs:WhileLoopSimplifier (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
while_loop_simplifier.h 33 class WhileLoopSimplifier : public HloPassInterface {
35 ~WhileLoopSimplifier() override {}
while_loop_simplifier_test.cc 109 ASSERT_TRUE(WhileLoopSimplifier().Run(&module()).ValueOrDie());
116 ASSERT_TRUE(WhileLoopSimplifier().Run(&module()).ValueOrDie());
123 EXPECT_FALSE(WhileLoopSimplifier().Run(&module()).ValueOrDie());
134 ASSERT_TRUE(WhileLoopSimplifier().Run(&module()).ValueOrDie());
152 EXPECT_FALSE(WhileLoopSimplifier().Run(&module()).ValueOrDie());
164 EXPECT_FALSE(WhileLoopSimplifier().Run(&module()).ValueOrDie());
178 EXPECT_FALSE(WhileLoopSimplifier().Run(&module()).ValueOrDie());
215 EXPECT_FALSE(WhileLoopSimplifier().Run(&module()).ValueOrDie());
251 EXPECT_FALSE(WhileLoopSimplifier().Run(&module()).ValueOrDie());
281 EXPECT_FALSE(WhileLoopSimplifier().Run(&module()).ValueOrDie())
    [all...]
while_loop_simplifier.cc 608 StatusOr<bool> WhileLoopSimplifier::Run(HloModule* module) {
610 "WhileLoopSimplifier::Run(), before:\n" + module->ToString());
653 "WhileLoopSimplifier::Run(), after:\n" + module->ToString());
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_compiler.cc 175 pass.AddPass<WhileLoopSimplifier>();
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_compiler.cc 274 pass.AddPass<WhileLoopSimplifier>();
    [all...]

Completed in 223 milliseconds