Lines Matching refs:constraint
270 TensorRankMutator(const TensorRankConstraint& constraint, in TensorRankMutator() argument
272 : mApplyToIndexes(applyToIndexes.begin(), applyToIndexes.end()), mConstraint(constraint) {}
720 auto constraint = TensorRankConstraint::Exactly(3); in TEST() local
721 auto validMutationSet = constraint.MutationsWithValidRank({operand}); in TEST()
748 auto constraint = TensorRankConstraint::Exactly(3); in TEST() local
749 auto invalidMutations = constraint.MutationsWithInvalidRank({operand}); in TEST()
774 auto constraint = TensorRankConstraint::AtLeast(1); in TEST() local
776 constraint.MutationsWithValidRank({(OperandTypeWithExtraParams)operand}); in TEST()
800 auto constraint = TensorRankConstraint::AtLeast(2); in TEST() local
802 constraint.MutationsWithInvalidRank({(OperandTypeWithExtraParams)operand}); in TEST()
818 auto constraint = TensorRankConstraint::AtLeast(1); in TEST() local
820 constraint.MutationsWithInvalidRank({(OperandTypeWithExtraParams)operand}); in TEST()
832 auto constraint = TensorRankConstraint::UpTo(3); in TEST() local
834 constraint.MutationsWithValidRank({(OperandTypeWithExtraParams)operand}); in TEST()
860 auto constraint = TensorRankConstraint::UpTo(3); in TEST() local
862 constraint.MutationsWithInvalidRank({(OperandTypeWithExtraParams)operand}); in TEST()
878 auto constraint = TensorRankConstraint::Between(2, 4); in TEST() local
879 auto validMutations = constraint.MutationsWithValidRank({(OperandTypeWithExtraParams)operand}); in TEST()
904 auto constraint = TensorRankConstraint::Between(2, 4); in TEST() local
906 constraint.MutationsWithInvalidRank({(OperandTypeWithExtraParams)operand}); in TEST()
931 auto constraint = TensorRankConstraint::Between(1, 4); in TEST() local
933 constraint.MutationsWithInvalidRank({(OperandTypeWithExtraParams)operand}); in TEST()