Searched refs:s32_shape (Results 1 – 1 of 1) sorted by relevance
279 Shape s32_shape = ShapeUtil::MakeShape(BF16, {1024}); in TEST_F() local284 HloInstruction::CreateParameter(1, s32_shape, "value")); in TEST_F()288 MakeSortHlo(ShapeUtil::MakeTupleShape({bf16_shape, s32_shape}), in TEST_F()