Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_simplifier_test.cc424 HloInstruction* new_while_op = in TEST_F() local
432 ShapeUtil::Equal(new_while_op->shape(), in TEST_F()
434 << ShapeUtil::HumanString(new_while_op->shape()); in TEST_F()
436 new_while_op->while_body()->root_instruction(), in TEST_F()
442 EXPECT_THAT(new_while_op->while_condition()->root_instruction(), in TEST_F()
479 HloInstruction* new_while_op = in TEST_F() local
484 EXPECT_TRUE(ShapeUtil::IsEmptyTuple(new_while_op->shape())) in TEST_F()
485 << new_while_op->shape().ToString(); in TEST_F()
Dwhile_loop_simplifier.cc160 auto* new_while_op = computation->AddInstruction(HloInstruction::CreateWhile( in RemoveDeadTupleIndices() local
191 new_while_op->shape().tuple_shapes(gte_idx), new_while_op, in RemoveDeadTupleIndices()
203 return new_while_op; in RemoveDeadTupleIndices()
1278 TF_ASSIGN_OR_RETURN(auto* new_while_op, in Run()
1280 if (new_while_op) { in Run()
1281 while_op = new_while_op; in Run()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dwhile_loop_outline.cc256 auto new_while_op = OpBuilder(while_op).create<WhileOp>( in OutlineWhile() local
258 new_while_op.cond().takeBody(while_op.cond()); in OutlineWhile()
259 new_while_op.body().takeBody(while_op.body()); in OutlineWhile()
261 new_while_op.getResults().take_front(while_op.getNumResults())); in OutlineWhile()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc3041 auto new_while_op = in matchAndRewrite() local
3046 rewriter.inlineRegionBefore(while_op.cond(), new_while_op.cond(), in matchAndRewrite()
3047 new_while_op.cond().end()); in matchAndRewrite()
3048 rewriter.inlineRegionBefore(while_op.body(), new_while_op.body(), in matchAndRewrite()
3049 new_while_op.body().end()); in matchAndRewrite()
3051 auto &new_cond_block = new_while_op.cond().front(); in matchAndRewrite()
3052 auto &new_body_block = new_while_op.body().front(); in matchAndRewrite()
3066 new_results.push_back(new_while_op.getResult(next_idx++)); in matchAndRewrite()