Lines Matching refs:ModulesTest
39 struct ModulesTest : torch::test::SeedingFixture {}; struct
41 TEST_F(ModulesTest, Conv1d) { in TEST_F() argument
61 TEST_F(ModulesTest, Conv1dSameStrided) { in TEST_F() argument
70 TEST_F(ModulesTest, Conv1dIvalidArg) { in TEST_F() argument
76 TEST_F(ModulesTest, Conv2dEven) { in TEST_F() argument
100 TEST_F(ModulesTest, Conv2dUneven) { in TEST_F() argument
122 TEST_F(ModulesTest, Conv2dSameStrided) { in TEST_F() argument
136 TEST_F(ModulesTest, Conv3d) { in TEST_F() argument
176 TEST_F(ModulesTest, Conv3dSameStrided) { in TEST_F() argument
190 TEST_F(ModulesTest, ConvTranspose1d) { in TEST_F() argument
210 TEST_F(ModulesTest, ConvTranspose2dEven) { in TEST_F() argument
245 TEST_F(ModulesTest, ConvTranspose2dUneven) { in TEST_F() argument
281 TEST_F(ModulesTest, ConvTranspose3d) { in TEST_F() argument
301 TEST_F(ModulesTest, MaxPool1d) { in TEST_F() argument
314 TEST_F(ModulesTest, MaxPool1dReturnIndices) { in TEST_F() argument
328 TEST_F(ModulesTest, MaxPool2dEven) { in TEST_F() argument
341 TEST_F(ModulesTest, MaxPool2dUneven) { in TEST_F() argument
354 TEST_F(ModulesTest, MaxPool2dReturnIndices) { in TEST_F() argument
368 TEST_F(ModulesTest, MaxPool3d) { in TEST_F() argument
381 TEST_F(ModulesTest, MaxPool3dReturnIndices) { in TEST_F() argument
399 TEST_F(ModulesTest, AvgPool1d) { in TEST_F() argument
412 TEST_F(ModulesTest, AvgPool2dEven) { in TEST_F() argument
425 TEST_F(ModulesTest, AvgPool2dUneven) { in TEST_F() argument
438 TEST_F(ModulesTest, AvgPool3d) { in TEST_F() argument
451 TEST_F(ModulesTest, FractionalMaxPool2d) { in TEST_F() argument
464 TEST_F(ModulesTest, FractionalMaxPool2dReturnIndices) { in TEST_F() argument
477 TEST_F(ModulesTest, FractionalMaxPool3d) { in TEST_F() argument
490 TEST_F(ModulesTest, FractionalMaxPool3dReturnIndices) { in TEST_F() argument
507 TEST_F(ModulesTest, LPPool1d) { in TEST_F() argument
525 TEST_F(ModulesTest, LPPool2d) { in TEST_F() argument
543 TEST_F(ModulesTest, LPPool3d) { in TEST_F() argument
561 TEST_F(ModulesTest, Identity) { in TEST_F() argument
574 TEST_F(ModulesTest, Flatten) { in TEST_F() argument
605 TEST_F(ModulesTest, Unflatten) { in TEST_F() argument
635 TEST_F(ModulesTest, AdaptiveMaxPool1d) { in TEST_F() argument
649 TEST_F(ModulesTest, AdaptiveMaxPool1dReturnIndices) { in TEST_F() argument
663 TEST_F(ModulesTest, AdaptiveMaxPool2dEven) { in TEST_F() argument
684 TEST_F(ModulesTest, AdaptiveMaxPool2dUneven) { in TEST_F() argument
705 TEST_F(ModulesTest, AdaptiveMaxPool2dReturnIndicesEven) { in TEST_F() argument
738 TEST_F(ModulesTest, AdaptiveMaxPool2dReturnIndicesUneven) { in TEST_F() argument
771 TEST_F(ModulesTest, AdaptiveMaxPool3d) { in TEST_F() argument
794 TEST_F(ModulesTest, AdaptiveMaxPool3dReturnIndices) { in TEST_F() argument
829 TEST_F(ModulesTest, AdaptiveAvgPool1d) { in TEST_F() argument
845 TEST_F(ModulesTest, AdaptiveAvgPool2dEven) { in TEST_F() argument
867 TEST_F(ModulesTest, AdaptiveAvgPool2dUneven) { in TEST_F() argument
889 TEST_F(ModulesTest, AdaptiveAvgPool3d) { in TEST_F() argument
912 TEST_F(ModulesTest, MaxUnpool1d) { in TEST_F() argument
936 TEST_F(ModulesTest, MaxPool1d_MaxUnpool1d) { in TEST_F() argument
956 TEST_F(ModulesTest, MaxUnpool2d) { in TEST_F() argument
986 TEST_F(ModulesTest, MaxPool2d_MaxUnpool2d) { in TEST_F() argument
1010 TEST_F(ModulesTest, MaxUnpool3d) { in TEST_F() argument
1028 TEST_F(ModulesTest, MaxUnpool3dOutputSize) { in TEST_F() argument
1049 TEST_F(ModulesTest, MaxPool3d_MaxUnpool3d) { in TEST_F() argument
1059 TEST_F(ModulesTest, Linear) { in TEST_F() argument
1096 TEST_F(ModulesTest, LocalResponseNorm) { in TEST_F() argument
1125 TEST_F(ModulesTest, LayerNorm) { in TEST_F() argument
1143 TEST_F(ModulesTest, GroupNorm) { in TEST_F() argument
1161 TEST_F(ModulesTest, Bilinear) { in TEST_F() argument
1177 TEST_F(ModulesTest, Fold) { in TEST_F() argument
1203 TEST_F(ModulesTest, Unfold) { in TEST_F() argument
1244 TEST_F(ModulesTest, SimpleContainer) { in TEST_F() argument
1262 TEST_F(ModulesTest, EmbeddingBasic) { in TEST_F() argument
1285 TEST_F(ModulesTest, EmbeddingList) { in TEST_F() argument
1298 TEST_F(ModulesTest, EmbeddingFromPretrained) { in TEST_F() argument
1306 TEST_F(ModulesTest, EmbeddingBagFromPretrained) { in TEST_F() argument
1315 TEST_F(ModulesTest, AlphaDropout) { in TEST_F() argument
1333 TEST_F(ModulesTest, FeatureAlphaDropout) { in TEST_F() argument
1352 TEST_F(ModulesTest, Dropout) { in TEST_F() argument
1377 TEST_F(ModulesTest, Dropout2d) { in TEST_F() argument
1406 TEST_F(ModulesTest, Dropout3d) { in TEST_F() argument
1436 TEST_F(ModulesTest, Parameters) { in TEST_F() argument
1456 TEST_F(ModulesTest, FunctionalCallsSuppliedFunction) { in TEST_F() argument
1473 TEST_F(ModulesTest, FunctionalWithTorchFunction) { in TEST_F() argument
1480 TEST_F(ModulesTest, FunctionalArgumentBinding) { in TEST_F() argument
1486 TEST_F(ModulesTest, BatchNorm1dStateful) { in TEST_F() argument
1513 TEST_F(ModulesTest, BatchNorm1dStateless) { in TEST_F() argument
1524 TEST_F(ModulesTest, BatchNorm1d) { in TEST_F() argument
1548 TEST_F(ModulesTest, BatchNorm2dStateful) { in TEST_F() argument
1575 TEST_F(ModulesTest, BatchNorm2dStateless) { in TEST_F() argument
1586 TEST_F(ModulesTest, BatchNorm2d) { in TEST_F() argument
1611 TEST_F(ModulesTest, BatchNorm3dStateful) { in TEST_F() argument
1638 TEST_F(ModulesTest, BatchNorm3dStateless) { in TEST_F() argument
1649 TEST_F(ModulesTest, BatchNorm3d) { in TEST_F() argument
1684 TEST_F(ModulesTest, InstanceNorm1dStateful) { in TEST_F() argument
1712 TEST_F(ModulesTest, InstanceNorm1dStateless) { in TEST_F() argument
1723 TEST_F(ModulesTest, InstanceNorm1d) { in TEST_F() argument
1747 TEST_F(ModulesTest, InstanceNorm2dStateful) { in TEST_F() argument
1775 TEST_F(ModulesTest, InstanceNorm2dStateless) { in TEST_F() argument
1786 TEST_F(ModulesTest, InstanceNorm2d) { in TEST_F() argument
1811 TEST_F(ModulesTest, InstanceNorm3dStateful) { in TEST_F() argument
1839 TEST_F(ModulesTest, InstanceNorm3dStateless) { in TEST_F() argument
1850 TEST_F(ModulesTest, InstanceNorm3d) { in TEST_F() argument
1885 TEST_F(ModulesTest, Linear_CUDA) { in TEST_F() argument
1902 TEST_F(ModulesTest, Linear2_CUDA) { in TEST_F() argument
1919 TEST_F(ModulesTest, L1Loss) { in TEST_F() argument
1931 TEST_F(ModulesTest, MSELoss) { in TEST_F() argument
1943 TEST_F(ModulesTest, BCELoss) { in TEST_F() argument
1955 TEST_F(ModulesTest, KLDivLoss) { in TEST_F() argument
1967 TEST_F(ModulesTest, HingeEmbeddingLoss) { in TEST_F() argument
1982 TEST_F(ModulesTest, MultiMarginLoss) { in TEST_F() argument
1998 TEST_F(ModulesTest, CosineEmbeddingLoss) { in TEST_F() argument
2015 TEST_F(ModulesTest, SmoothL1LossDefaultOptions) { in TEST_F() argument
2029 TEST_F(ModulesTest, HuberLossDefaultOptions) { in TEST_F() argument
2043 TEST_F(ModulesTest, MultiLabelMarginLossDefaultOptions) { in TEST_F() argument
2057 TEST_F(ModulesTest, SmoothL1LossNoReduction) { in TEST_F() argument
2071 TEST_F(ModulesTest, HuberLossNoReduction) { in TEST_F() argument
2085 TEST_F(ModulesTest, MultiLabelMarginLossNoReduction) { in TEST_F() argument
2099 TEST_F(ModulesTest, SmoothL1LossBeta) { in TEST_F() argument
2114 TEST_F(ModulesTest, HuberLossDelta) { in TEST_F() argument
2129 TEST_F(ModulesTest, TripletMarginLoss) { in TEST_F() argument
2146 TEST_F(ModulesTest, TripletMarginWithDistanceLossDefaultParity) { in TEST_F() argument
2198 TEST_F(ModulesTest, TripletMarginWithDistanceLossFunctionalParity) { in TEST_F() argument
2255 TEST_F(ModulesTest, NLLLoss) { in TEST_F() argument
2275 TEST_F(ModulesTest, CrossEntropyLoss) { in TEST_F() argument
2322 TEST_F(ModulesTest, CosineSimilarity) { in TEST_F() argument
2337 TEST_F(ModulesTest, SoftMarginLossDefaultOptions) { in TEST_F() argument
2351 TEST_F(ModulesTest, MultiLabelSoftMarginLossDefaultOptions) { in TEST_F() argument
2367 TEST_F(ModulesTest, SoftMarginLossNoReduction) { in TEST_F() argument
2382 TEST_F(ModulesTest, MultiLabelSoftMarginLossWeightedNoReduction) { in TEST_F() argument
2401 TEST_F(ModulesTest, PairwiseDistance) { in TEST_F() argument
2416 TEST_F(ModulesTest, ELU) { in TEST_F() argument
2447 TEST_F(ModulesTest, SELU) { in TEST_F() argument
2474 TEST_F(ModulesTest, Hardshrink) { in TEST_F() argument
2492 TEST_F(ModulesTest, Hardtanh) { in TEST_F() argument
2526 TEST_F(ModulesTest, HardtanhMinValGEMaxVal) { in TEST_F() argument
2541 TEST_F(ModulesTest, LeakyReLU) { in TEST_F() argument
2573 TEST_F(ModulesTest, LogSigmoid) { in TEST_F() argument
2591 TEST_F(ModulesTest, Softmax) { in TEST_F() argument
2603 TEST_F(ModulesTest, Softmin) { in TEST_F() argument
2615 TEST_F(ModulesTest, LogSoftmax) { in TEST_F() argument
2627 TEST_F(ModulesTest, AdaptiveLogSoftmaxWithLoss) { in TEST_F() argument
2702 TEST_F(ModulesTest, Softmax2d) { in TEST_F() argument
2720 TEST_F(ModulesTest, PReLU) { in TEST_F() argument
2743 TEST_F(ModulesTest, ReLU) { in TEST_F() argument
2769 TEST_F(ModulesTest, ReLU6) { in TEST_F() argument
2796 TEST_F(ModulesTest, RReLU) { in TEST_F() argument
2832 TEST_F(ModulesTest, CELU) { in TEST_F() argument
2862 TEST_F(ModulesTest, GLU) { in TEST_F() argument
2881 TEST_F(ModulesTest, GELU) { in TEST_F() argument
2889 TEST_F(ModulesTest, TanhGELU) { in TEST_F() argument
2899 TEST_F(ModulesTest, Mish) { in TEST_F() argument
2908 TEST_F(ModulesTest, Sigmoid) { in TEST_F() argument
2917 TEST_F(ModulesTest, PixelShuffle) { in TEST_F() argument
2935 TEST_F(ModulesTest, PixelUnshuffle) { in TEST_F() argument
2953 TEST_F(ModulesTest, Softplus) { in TEST_F() argument
2972 TEST_F(ModulesTest, Softshrink) { in TEST_F() argument
2991 TEST_F(ModulesTest, Softsign) { in TEST_F() argument
3000 TEST_F(ModulesTest, Tanh) { in TEST_F() argument
3009 TEST_F(ModulesTest, Tanhshrink) { in TEST_F() argument
3018 TEST_F(ModulesTest, Threshold) { in TEST_F() argument
3042 TEST_F(ModulesTest, Upsampling1D) { in TEST_F() argument
3092 TEST_F(ModulesTest, Upsampling2D) { in TEST_F() argument
3152 TEST_F(ModulesTest, Upsampling3D) { in TEST_F() argument
3191 TEST_F(ModulesTest, CTCLoss) { in TEST_F() argument
3206 TEST_F(ModulesTest, PoissonNLLLoss) { in TEST_F() argument
3227 TEST_F(ModulesTest, MarginRankingLoss) { in TEST_F() argument
3261 TEST_F(ModulesTest, BCEWithLogitsLoss) { in TEST_F() argument
3777 TEST_F(ModulesTest, MultiheadAttention) { in TEST_F() argument
3860 TEST_F(ModulesTest, PrettyPrintIdentity) { in TEST_F() argument
3864 TEST_F(ModulesTest, PrettyPrintFlatten) { in TEST_F() argument
3871 TEST_F(ModulesTest, PrettyPrintUnflatten) { in TEST_F() argument
3883 TEST_F(ModulesTest, ReflectionPad1d) { in TEST_F() argument
3904 TEST_F(ModulesTest, ReflectionPad2d) { in TEST_F() argument
3935 TEST_F(ModulesTest, ReflectionPad3d) { in TEST_F() argument
3977 TEST_F(ModulesTest, ReplicationPad1d) { in TEST_F() argument
3998 TEST_F(ModulesTest, ReplicationPad2d) { in TEST_F() argument
4029 TEST_F(ModulesTest, ReplicationPad3d) { in TEST_F() argument
4089 TEST_F(ModulesTest, ZeroPad1d) { in TEST_F() argument
4110 TEST_F(ModulesTest, ZeroPad2d) { in TEST_F() argument
4141 TEST_F(ModulesTest, ZeroPad3d) { in TEST_F() argument
4201 TEST_F(ModulesTest, ConstantPad1d) { in TEST_F() argument
4224 TEST_F(ModulesTest, ConstantPad2d) { in TEST_F() argument
4254 TEST_F(ModulesTest, ConstantPad3d) { in TEST_F() argument
4314 TEST_F(ModulesTest, CrossMapLRN2d) { in TEST_F() argument
4380 TEST_F(ModulesTest, RNNCell) { in TEST_F() argument
4432 TEST_F(ModulesTest, LSTMCell) { in TEST_F() argument
4525 TEST_F(ModulesTest, GRUCell) { in TEST_F() argument
4577 TEST_F(ModulesTest, PrettyPrintLinear) { in TEST_F() argument
4583 TEST_F(ModulesTest, PrettyPrintBilinear) { in TEST_F() argument
4592 TEST_F(ModulesTest, PrettyPrintConv) { in TEST_F() argument
4637 TEST_F(ModulesTest, PrettyPrintConvTranspose) { in TEST_F() argument
4683 TEST_F(ModulesTest, PrettyPrintUpsample) { in TEST_F() argument
4695 TEST_F(ModulesTest, PrettyPrintFold) { in TEST_F() argument
4705 TEST_F(ModulesTest, PrettyPrintUnfold) { in TEST_F() argument
4715 TEST_F(ModulesTest, PrettyPrintMaxPool) { in TEST_F() argument
4739 TEST_F(ModulesTest, PrettyPrintAvgPool) { in TEST_F() argument
4763 TEST_F(ModulesTest, PrettyPrinFractionalMaxPool) { in TEST_F() argument
4774 TEST_F(ModulesTest, PrettyPrintLPPool) { in TEST_F() argument
4799 TEST_F(ModulesTest, PrettyPrintAdaptiveMaxPool) { in TEST_F() argument
4839 TEST_F(ModulesTest, PrettyPrintAdaptiveAvgPool) { in TEST_F() argument
4874 TEST_F(ModulesTest, PrettyPrintMaxUnpool) { in TEST_F() argument
4895 TEST_F(ModulesTest, PrettyPrintDropout) { in TEST_F() argument
4904 TEST_F(ModulesTest, PrettyPrintDropout2d) { in TEST_F() argument
4914 TEST_F(ModulesTest, PrettyPrintDropout3d) { in TEST_F() argument
4924 TEST_F(ModulesTest, PrettyPrintFunctional) { in TEST_F() argument
4928 TEST_F(ModulesTest, PrettyPrintBatchNorm1d) { in TEST_F() argument
4938 TEST_F(ModulesTest, PrettyPrintBatchNorm2d) { in TEST_F() argument
4948 TEST_F(ModulesTest, PrettyPrintBatchNorm3d) { in TEST_F() argument
4958 TEST_F(ModulesTest, PrettyPrintInstanceNorm1d) { in TEST_F() argument
4968 TEST_F(ModulesTest, PrettyPrintInstanceNorm2d) { in TEST_F() argument
4978 TEST_F(ModulesTest, PrettyPrintInstanceNorm3d) { in TEST_F() argument
4988 TEST_F(ModulesTest, PrettyPrintLayerNorm) { in TEST_F() argument
4998 TEST_F(ModulesTest, PrettyPrintGroupNorm) { in TEST_F() argument
5007 TEST_F(ModulesTest, PrettyPrintLocalResponseNorm) { in TEST_F() argument
5017 TEST_F(ModulesTest, PrettyPrintEmbedding) { in TEST_F() argument
5034 TEST_F(ModulesTest, PrettyPrintEmbeddingBag) { in TEST_F() argument
5067 TEST_F(ModulesTest, PrettyPrintL1Loss) { in TEST_F() argument
5070 TEST_F(ModulesTest, PrettyPrintKLDivLoss) { in TEST_F() argument
5073 TEST_F(ModulesTest, PrettyPrintMSELoss) { in TEST_F() argument
5076 TEST_F(ModulesTest, PrettyPrintBCELoss) { in TEST_F() argument
5079 TEST_F(ModulesTest, PrettyPrintHingeEmbeddingLoss) { in TEST_F() argument
5085 TEST_F(ModulesTest, PrettyPrintCosineEmbeddingLoss) { in TEST_F() argument
5091 TEST_F(ModulesTest, PrettyPrintTripletMarginLoss) { in TEST_F() argument
5098 TEST_F(ModulesTest, PrettyPrintTripletMarginWithDistanceLoss) { in TEST_F() argument
5112 TEST_F(ModulesTest, PrettyPrintNLLLoss) { in TEST_F() argument
5116 TEST_F(ModulesTest, PrettyPrinCrossEntropyLoss) { in TEST_F() argument
5120 TEST_F(ModulesTest, PrettyPrintMultiLabelMarginLoss) { in TEST_F() argument
5125 TEST_F(ModulesTest, PrettyPrintMultiLabelSoftMarginLoss) { in TEST_F() argument
5131 TEST_F(ModulesTest, PrettyPrintSoftMarginLoss) { in TEST_F() argument
5135 TEST_F(ModulesTest, PrettyPrintCosineSimilarity) { in TEST_F() argument
5144 TEST_F(ModulesTest, PrettyPrintPairwiseDistance) { in TEST_F() argument
5154 TEST_F(ModulesTest, PrettyPrintReflectionPad) { in TEST_F() argument
5169 TEST_F(ModulesTest, PrettyPrintReplicationPad) { in TEST_F() argument
5190 TEST_F(ModulesTest, PrettyPrintZeroPad) { in TEST_F() argument
5211 TEST_F(ModulesTest, PrettyPrintConstantPad) { in TEST_F() argument
5232 TEST_F(ModulesTest, PrettyPrintNestedModel) { in TEST_F() argument
5270 TEST_F(ModulesTest, PrettyPrintELU) { in TEST_F() argument
5277 TEST_F(ModulesTest, PrettyPrintSELU) { in TEST_F() argument
5284 TEST_F(ModulesTest, PrettyPrintGLU) { in TEST_F() argument
5289 TEST_F(ModulesTest, PrettyPrintHardshrink) { in TEST_F() argument
5296 TEST_F(ModulesTest, PrettyPrintHardtanh) { in TEST_F() argument
5304 TEST_F(ModulesTest, PrettyPrintLeakyReLU) { in TEST_F() argument
5312 TEST_F(ModulesTest, PrettyPrintLogSigmoid) { in TEST_F() argument
5316 TEST_F(ModulesTest, PrettyPrintSoftmax) { in TEST_F() argument
5320 TEST_F(ModulesTest, PrettyPrintSoftmin) { in TEST_F() argument
5324 TEST_F(ModulesTest, PrettyPrintLogSoftmax) { in TEST_F() argument
5330 TEST_F(ModulesTest, PrettyPrintSoftmax2d) { in TEST_F() argument
5334 TEST_F(ModulesTest, PrettyPrintPReLU) { in TEST_F() argument
5341 TEST_F(ModulesTest, PrettyPrintReLU) { in TEST_F() argument
5349 TEST_F(ModulesTest, PrettyPrintReLU6) { in TEST_F() argument
5358 TEST_F(ModulesTest, PrettyPrintRReLU) { in TEST_F() argument
5365 TEST_F(ModulesTest, PrettyPrintCELU) { in TEST_F() argument
5372 TEST_F(ModulesTest, PrettyPrintSigmoid) { in TEST_F() argument
5376 TEST_F(ModulesTest, PrettyPrintPixelShuffle) { in TEST_F() argument
5382 TEST_F(ModulesTest, PrettyPrintPixelUnshuffle) { in TEST_F() argument
5388 TEST_F(ModulesTest, PrettyPrintSoftplus) { in TEST_F() argument
5395 TEST_F(ModulesTest, PrettyPrintSoftshrink) { in TEST_F() argument
5402 TEST_F(ModulesTest, PrettyPrintSoftsign) { in TEST_F() argument
5406 TEST_F(ModulesTest, PrettyPrintTanh) { in TEST_F() argument
5410 TEST_F(ModulesTest, PrettyPrintTanhshrink) { in TEST_F() argument
5414 TEST_F(ModulesTest, PrettyPrintThreshold) { in TEST_F() argument
5423 TEST_F(ModulesTest, PrettyPrintCTCLoss) { in TEST_F() argument
5432 TEST_F(ModulesTest, PrettyPrintPoissonNLLLoss) { in TEST_F() argument
5443 TEST_F(ModulesTest, PrettyPrintMarginRankingLoss) { in TEST_F() argument
5451 TEST_F(ModulesTest, PrettyPrintCrossMapLRN2d) { in TEST_F() argument
5461 TEST_F(ModulesTest, PrettyPrintAlphaDropout) { in TEST_F() argument
5473 TEST_F(ModulesTest, PrettyPrintFeatureAlphaDropout) { in TEST_F() argument
5486 TEST_F(ModulesTest, PrettyPrintBCEWithLogitsLoss) { in TEST_F() argument
5496 TEST_F(ModulesTest, PrettyPrintMultiheadAttention) { in TEST_F() argument
5506 TEST_F(ModulesTest, PrettyPrintRNNCell) { in TEST_F() argument
5518 TEST_F(ModulesTest, PrettyPrintLSTMCell) { in TEST_F() argument
5525 TEST_F(ModulesTest, PrettyPrintGRUCell) { in TEST_F() argument
5532 TEST_F(ModulesTest, PrettyPrintAdaptiveLogSoftmaxWithLoss) { in TEST_F() argument