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