Lines Matching refs:PRED
40 const Shape pred_ = ShapeUtil::MakeShape(PRED, {});
126 auto predarray = ShapeUtil::MakeShape(PRED, {64, 48});
144 HasSubstr("pred operand must have PRED"));
147 TernaryOperation::TRIOP_SELECT, ShapeUtil::MakeShape(PRED, {64}),
160 HasSubstr("pred operand must have PRED element type"));
365 HasSubstr("select function must have rank-0 PRED"));
818 ASSERT_TRUE(ShapeUtil::Equal(ShapeUtil::MakeShape(PRED, {10}),
827 ASSERT_TRUE(ShapeUtil::Equal(ShapeUtil::MakeShape(PRED, {10}),
836 ASSERT_TRUE(ShapeUtil::Equal(ShapeUtil::MakeShape(PRED, {10}),
845 ASSERT_TRUE(ShapeUtil::Equal(ShapeUtil::MakeShape(PRED, {10}),
854 ASSERT_TRUE(ShapeUtil::Equal(ShapeUtil::MakeShape(PRED, {10}),
863 ASSERT_TRUE(ShapeUtil::Equal(ShapeUtil::MakeShape(PRED, {10}),