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.cc696 Shape new_while_shape = ParseShape("(s32[1], s32[3])").ValueOrDie(); in TEST_F() local
697 EXPECT_TRUE(ShapeUtil::Equal(new_while->shape(), new_while_shape)); in TEST_F()
699 new_while->while_body()->root_instruction()->shape(), new_while_shape)); in TEST_F()
702 new_while_shape)); in TEST_F()
705 new_while_shape)); in TEST_F()
765 Shape new_while_shape = in TEST_F() local
767 EXPECT_TRUE(ShapeUtil::Equal(new_while->shape(), new_while_shape)); in TEST_F()
769 new_while->while_body()->root_instruction()->shape(), new_while_shape)); in TEST_F()
772 new_while_shape)); in TEST_F()
775 new_while_shape)); in TEST_F()
[all …]
Dwhile_loop_simplifier.cc73 Shape new_while_shape = in RemoveDeadTupleIndices() local
85 0, new_while_shape, param->name())); in RemoveDeadTupleIndices()
161 new_while_shape, in RemoveDeadTupleIndices()
532 Shape new_while_shape = ShapeUtil::MakeTupleShape(new_while_shape_elems); in TryRemoveConstantParams() local
559 CHECK(ShapeUtil::Compatible(instr->shape(), new_while_shape)); in TryRemoveConstantParams()
586 if (ShapeUtil::IsEmptyTuple(new_while_shape)) { in TryRemoveConstantParams()
595 0, new_while_shape, in TryRemoveConstantParams()
604 0, new_while_shape, in TryRemoveConstantParams()
620 new_while_shape, in TryRemoveConstantParams()
1065 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()
136 i < new_while_shape.tuple_shapes_size(); i++) { in MakeInstructionsLiveIn()