Lines Matching refs:TensorRankConstraint
142 class TensorRankConstraint { class
144 TensorRankConstraint(const TensorRankConstraint& copyFrom) { in TensorRankConstraint() function in __anon37b312a30111::TensorRankConstraint
150 TensorRankConstraint& operator=(const TensorRankConstraint& copyFrom) { in operator =()
157 static TensorRankConstraint Exactly(uint32_t rank) { in Exactly()
158 return TensorRankConstraint(std::optional(rank), std::optional(rank)); in Exactly()
161 static TensorRankConstraint AtLeast(uint32_t min) { in AtLeast()
162 return TensorRankConstraint(std::optional(min), std::nullopt); in AtLeast()
165 static TensorRankConstraint UpTo(uint32_t max) { in UpTo()
166 return TensorRankConstraint(std::nullopt, std::optional(max)); in UpTo()
169 static TensorRankConstraint Between(uint32_t min, uint32_t max) { in Between()
173 return TensorRankConstraint(std::optional(min), std::optional(max)); in Between()
253 TensorRankConstraint(const std::optional<uint32_t>& min, const std::optional<uint32_t>& max) in TensorRankConstraint() function in __anon37b312a30111::TensorRankConstraint
270 TensorRankMutator(const TensorRankConstraint& constraint, in TensorRankMutator()
326 TensorRankConstraint mConstraint;
712 TEST(TensorRankConstraint, ExactlyWillReturnSameInputAsValidMutation) { in TEST() argument
720 auto constraint = TensorRankConstraint::Exactly(3); in TEST()
729 TEST(TensorRankConstraint, ExactlyWillFailIfValidInputHasInvalidSize) { in TEST() argument
736 EXPECT_DEATH(TensorRankConstraint::Exactly(3).MutationsWithValidRank({operand}), in TEST()
740 TEST(TensorRankConstraint, ExactlyWillReturnTwoInvalidMutationsWithLowerAndHigherRank) { in TEST() argument
748 auto constraint = TensorRankConstraint::Exactly(3); in TEST()
766 TEST(TensorRankConstraint, AtLeastWillReturnTwoValidMutationsAboveThreshold) { in TEST() argument
774 auto constraint = TensorRankConstraint::AtLeast(1); in TEST()
792 TEST(TensorRankConstraint, AtLeastWillReturnOneInvalidMutationsBelowThreshold) { in TEST() argument
800 auto constraint = TensorRankConstraint::AtLeast(2); in TEST()
810 TEST(TensorRankConstraint, AtLeastWillReturnNoInvalidMutationsIfThresholdIs1) { in TEST() argument
818 auto constraint = TensorRankConstraint::AtLeast(1); in TEST()
824 TEST(TensorRankConstraint, UpToWillReturnUpToTwoValidMutationsBelowThreshold) { in TEST() argument
832 auto constraint = TensorRankConstraint::UpTo(3); in TEST()
852 TEST(TensorRankConstraint, UpToWillReturnOneInvalidMutationsAboveThreshold) { in TEST() argument
860 auto constraint = TensorRankConstraint::UpTo(3); in TEST()
870 TEST(TensorRankConstraint, BetweenWillReturnTwoValidMutationsOnRangeBoundaries) { in TEST() argument
878 auto constraint = TensorRankConstraint::Between(2, 4); in TEST()
896 TEST(TensorRankConstraint, BetweenWillReturnTwoInvValidMutationsAdjacentToRangeBoundaries) { in TEST() argument
904 auto constraint = TensorRankConstraint::Between(2, 4); in TEST()
923 TEST(TensorRankConstraint, BetweenWillReturnOneInvalidMutationsOnlyIfLowerBoundIs1) { in TEST() argument
931 auto constraint = TensorRankConstraint::Between(1, 4); in TEST()
960 TensorRankMutator mutator{TensorRankConstraint::AtLeast(2), {0, 2}}; in TEST()
1008 TensorRankMutator mutator{TensorRankConstraint::AtLeast(2), {0, 2}}; in TEST()
1070 {{TensorRankConstraint::UpTo(4)}}); in dequantizeOpTest()
1304 {{TensorRankConstraint::UpTo(4), {0}}, {TensorRankConstraint::UpTo(4), {1}}}); in simpleMathOpTest()
1459 inputRankMutators.push_back({TensorRankConstraint::UpTo(4)}); in activationOpTest()
1666 {{TensorRankConstraint::UpTo(4)}}); in reshapeOpTest()
1723 {{TensorRankConstraint::UpTo(4)}}); in meanOpTest()
1753 {{TensorRankConstraint::UpTo(4)}}); in padOpTest()
1781 {{TensorRankConstraint::UpTo(4)}}); in padV2OpTest()
1803 {{TensorRankConstraint::UpTo(4)}}); in softmaxOpTest()
1808 {{TensorRankConstraint::UpTo(4)}}); in softmaxOpTest()
2055 {{TensorRankConstraint::UpTo(4)}}); in transposeAndSqueezeOpTest()
2152 {{TensorRankConstraint::Exactly(4), {0, 1}}}); in convOpTest()
2400 {{TensorRankConstraint::Between(2, 4), {0}}, in fullyConnectedOpTest()
2401 {TensorRankConstraint::Exactly(2), {1}}, in fullyConnectedOpTest()
2402 {TensorRankConstraint::Exactly(1), {2}}}); in fullyConnectedOpTest()
2435 {output}, {{TensorRankConstraint::UpTo(4), {0, 1}}}); in concatenationTest()
2439 {{TensorRankConstraint::UpTo(4)}}); in concatenationTest()
3169 {{TensorRankConstraint::UpTo(4)}}); in stridedSliceOpTest()
3468 {getOpType(operandCode, 2, inoutDim)}, {{TensorRankConstraint::UpTo(4)}}); in channelShuffleOpTest()
3569 {getOpType(operandCode, 4, inputDim)}, {{TensorRankConstraint::UpTo(4)}}); in normalizationOpTest()
3599 {getOpType(operandCode, 4, inputDim)}, {{TensorRankConstraint::UpTo(4), {0}}}); in localResponseNormOpTest()
3607 {getOpType(operandCode, 4, inputDim)}, {{TensorRankConstraint::UpTo(4), {0}}}); in localResponseNormOpTest()
3776 {{TensorRankConstraint::UpTo(4)}}); in reduceOpTest()