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
100 const Shape& while_shape = while_instr->shape(); in TEST_F() local
421 auto& while_shape = while_init->shape(); in TryRemoveRepeatedWhileTupleIndices() local509 const auto& while_shape = while_init->shape(); in TryRemoveConstantParams() local870 Shape while_shape = while_init->shape(); in TryFlattenNestedTuples() local997 Shape while_shape = while_init->shape(); in TryMergeInductionVariables() local
378 Shape while_shape = ShapeUtil::MakeTupleShape(parameter_shapes); in XLA_TEST_F() local
1192 auto while_shape = ShapeUtil::MakeTupleShape( in XLA_TEST_F() local1234 auto while_shape = ShapeUtil::MakeShape(S32, {}); in XLA_TEST_F() local