Home
last modified time | relevance | path

Searched refs:new_while_shape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_simplifier_test.cc642 Shape new_while_shape = ParseShape("(s32[1], s32[3])").ValueOrDie(); in TEST_F() local
643 EXPECT_TRUE(ShapeUtil::Equal(new_while->shape(), new_while_shape)); in TEST_F()
645 new_while->while_body()->root_instruction()->shape(), new_while_shape)); in TEST_F()
648 new_while_shape)); in TEST_F()
651 new_while_shape)); in TEST_F()
711 Shape new_while_shape = in TEST_F() local
713 EXPECT_TRUE(ShapeUtil::Equal(new_while->shape(), new_while_shape)); in TEST_F()
715 new_while->while_body()->root_instruction()->shape(), new_while_shape)); in TEST_F()
718 new_while_shape)); in TEST_F()
721 new_while_shape)); in TEST_F()
Dwhile_loop_simplifier.cc176 Shape new_while_shape = in TryRemoveDeadWhileParams() local
188 0, new_while_shape, param->name())); in TryRemoveDeadWhileParams()
264 new_while_shape, in TryRemoveDeadWhileParams()
351 Shape new_while_shape = ShapeUtil::MakeTupleShape(new_while_shape_elems); in TryRemoveConstantParams() local
378 CHECK(ShapeUtil::Compatible(instr->shape(), new_while_shape)); in TryRemoveConstantParams()
405 if (ShapeUtil::IsEmptyTuple(new_while_shape)) { in TryRemoveConstantParams()
414 0, new_while_shape, in TryRemoveConstantParams()
423 0, new_while_shape, in TryRemoveConstantParams()
439 new_while_shape, in TryRemoveConstantParams()
853 Shape new_while_shape = while_shape; in TryMergeInductionVariables() local
[all …]
Dwhile_util.cc103 Shape new_while_shape = while_instr->shape(); in MakeInstructionsLiveIn() local
105 *new_while_shape.add_tuple_shapes() = instruction->shape(); in MakeInstructionsLiveIn()
110 WidenWhileCondition(while_instr->while_condition(), new_while_shape)); in MakeInstructionsLiveIn()
116 WidenWhileBody(while_instr->while_body(), new_while_shape)); in MakeInstructionsLiveIn()
122 HloInstruction::CreateWhile(new_while_shape, new_while_condition, in MakeInstructionsLiveIn()
135 i < new_while_shape.tuple_shapes_size(); i++) { in MakeInstructionsLiveIn()