Searched defs:pred_shape (Results 1 – 3 of 3) 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
255 const Shape pred_shape = in PadWithScalar() local
2547 auto pred_shape = ShapeUtil::ChangeElementType(gather->shape(), PRED); in HandleGather() local