Searched defs:while_shape (Results 1 – 5 of 5) sorted by relevance
63 Shape while_shape = in TEST_F() local105 Shape while_shape = in TEST_F() local178 Shape while_shape = ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32}); in TEST_F() local216 Shape while_shape = in TEST_F() local255 Shape while_shape = in TEST_F() local304 Shape while_shape = in TEST_F() local358 Shape while_shape = ShapeUtil::MakeTupleShape( in TEST_F() local408 Shape while_shape = in TEST_F() local445 Shape while_shape = ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32}); in TEST_F() local
558 auto& while_shape = while_init->shape(); in TryRemoveRepeatedWhileTupleIndices() local646 const auto& while_shape = while_init->shape(); in TryRemoveConstantParams() local1013 Shape while_shape = while_init->shape(); in TryFlattenNestedTuples() local1144 Shape while_shape = while_init->shape(); in TryMergeInductionVariables() local
100 const Shape& while_shape = while_instr->shape(); in TEST_F() local
1178 auto while_shape = ShapeUtil::MakeTupleShape( in XLA_TEST_F() local1220 auto while_shape = ShapeUtil::MakeShape(S32, {}); in XLA_TEST_F() local
382 Shape while_shape = ShapeUtil::MakeTupleShape(parameter_shapes); in XLA_TEST_F() local