mindspore2022/tests/ut/python/nn
huangxinjing 18044aff0f 1. Add docstring, elimitate attention mask, tuple append the deocoder return layer past
split attention inputs from two into three

Add input check and modify test case

Fix check error and add check to transformer inputs

add input check to the transformers
2021-08-16 21:08:48 +08:00
..
optim optimize run memory 2021-08-10 20:45:25 +08:00
probability update doc example in probability 2021-01-19 11:51:43 -05:00
__init__.py pylint waring clean 2020-05-13 11:30:27 +08:00
test_activation.py add raises description for BCELoss, ReLU, BatchNorm1d, etc. operators. 2021-02-02 20:52:00 +08:00
test_batchnorm.py pylint waring clean 2020-05-13 11:30:27 +08:00
test_cell.py remove the 'raise' in construct of Cell 2020-12-17 10:54:42 +08:00
test_cell_wrapper.py implicit type conversion 2020-05-27 20:04:52 +08:00
test_central_crop.py Add image.CentralCrop 2020-06-16 16:10:17 +08:00
test_checkparameter.py fix CI random fails 2020-11-02 16:28:28 +08:00
test_clip_by_norm.py clean pylint 2020-06-04 21:03:30 +08:00
test_container.py set default context mode to GRAPH_MODE 2021-06-09 14:40:49 +08:00
test_conv.py !906 fix a bug that support dilation greater than 1 in conv2dTranspose ops 2020-05-09 21:58:27 +08:00
test_dense.py Extension interface for dense 2020-10-22 17:58:24 +08:00
test_dropout.py delete seed0 and seed1 in Dropout 2020-09-08 10:51:12 +08:00
test_dynamic_lr.py uniform learning_rate behavior of optimizers 2020-07-20 21:09:20 +08:00
test_flatten.py pylint waring clean 2020-05-13 11:30:27 +08:00
test_image_gradients.py pylint waring clean 2020-05-13 11:30:27 +08:00
test_l1_regularizer.py optimize log 2021-06-04 12:32:44 +08:00
test_learning_rate_schedule.py Check whether the network args are tensors in the compile phase 2020-10-19 11:46:10 +08:00
test_loss.py diceloss bug 2021-03-09 16:23:45 +08:00
test_lstm.py pylint waring clean 2020-05-13 11:30:27 +08:00
test_msssim.py update SSIM loss, add MSSSIM loss feature; add their ut testcases. 2020-07-10 11:11:48 +08:00
test_nn_embedding.py Add TopoSort Rhs First attribute for special CNode, such as Depend CNode with isolated nodes. 2021-03-06 12:39:36 +08:00
test_nn_pad.py remove name arg from gradoperation 2020-08-26 11:16:57 +08:00
test_norm.py pylint waring clean 2020-05-13 11:30:27 +08:00
test_parameter.py add complex number support 2021-08-12 10:36:43 +08:00
test_pooling.py clean pylint 2020-05-29 23:00:00 +08:00
test_psnr.py clean pylint 2020-05-29 23:00:00 +08:00
test_resizebilinear.py change name from interpolate to resizeBilinear 2020-12-10 11:46:40 +08:00
test_ssim.py Add TopoSort Rhs First attribute for special CNode, such as Depend CNode with isolated nodes. 2021-03-06 12:39:36 +08:00
test_transformer.py 1. Add docstring, elimitate attention mask, tuple append the deocoder return layer past 2021-08-16 21:08:48 +08:00
test_tril.py Add nn.Tril function 2020-11-19 10:31:30 +08:00
test_triu.py Add nn.Tril function 2020-11-19 10:31:30 +08:00