Searched defs:pred_shape (Results 1 – 6 of 6) sorted by relevance
421 auto pred_shape = ShapeUtil::MakeShape(PRED, {8, 7}); in TEST_F() local455 auto pred_shape = ShapeUtil::MakeShape(PRED, {}); in TEST_F() local
265 const Shape pred_shape = in PadWithScalar() local
2609 auto pred_shape = ShapeUtil::ChangeElementType(gather->shape(), PRED); in HandleGather() local
466 auto pred_shape = ShapeUtil::MakeShape(PRED, {}); in TEST_F() local
1244 Value pred_shape = rewriter.createOrFold<shape::ShapeOfOp>(loc, pred); in matchAndRewrite() local
4047 Shape pred_shape = ShapeUtil::ChangeElementType(hlo->shape(), PRED); in PreprocessHlos() local