Searched refs:xshape (Results 1 – 1 of 1) sorted by relevance
787 Shape xshape = ShapeUtil::MakeShapeWithLayout(F32, {128, 8}, {0, 1}); in TEST_F() local793 HloInstruction::CreateInfeed(xshape, token, "")); in TEST_F()795 HloInstruction::CreateGetTupleElement(xshape, infeed, 0)); in TEST_F()