Lines Matching refs:constraint
271 TensorRankMutator(const TensorRankConstraint& constraint, in TensorRankMutator() argument
273 : mApplyToIndexes(applyToIndexes.begin(), applyToIndexes.end()), mConstraint(constraint) {}
721 auto constraint = TensorRankConstraint::Exactly(3); in TEST() local
722 auto validMutationSet = constraint.MutationsWithValidRank({operand}); in TEST()
749 auto constraint = TensorRankConstraint::Exactly(3); in TEST() local
750 auto invalidMutations = constraint.MutationsWithInvalidRank({operand}); in TEST()
775 auto constraint = TensorRankConstraint::AtLeast(1); in TEST() local
777 constraint.MutationsWithValidRank({(OperandTypeWithExtraParams)operand}); in TEST()
801 auto constraint = TensorRankConstraint::AtLeast(2); in TEST() local
803 constraint.MutationsWithInvalidRank({(OperandTypeWithExtraParams)operand}); in TEST()
819 auto constraint = TensorRankConstraint::AtLeast(1); in TEST() local
821 constraint.MutationsWithInvalidRank({(OperandTypeWithExtraParams)operand}); in TEST()
833 auto constraint = TensorRankConstraint::UpTo(3); in TEST() local
835 constraint.MutationsWithValidRank({(OperandTypeWithExtraParams)operand}); in TEST()
861 auto constraint = TensorRankConstraint::UpTo(3); in TEST() local
863 constraint.MutationsWithInvalidRank({(OperandTypeWithExtraParams)operand}); in TEST()
879 auto constraint = TensorRankConstraint::Between(2, 4); in TEST() local
880 auto validMutations = constraint.MutationsWithValidRank({(OperandTypeWithExtraParams)operand}); in TEST()
905 auto constraint = TensorRankConstraint::Between(2, 4); in TEST() local
907 constraint.MutationsWithInvalidRank({(OperandTypeWithExtraParams)operand}); in TEST()
932 auto constraint = TensorRankConstraint::Between(1, 4); in TEST() local
934 constraint.MutationsWithInvalidRank({(OperandTypeWithExtraParams)operand}); in TEST()