mindspore/tests/ut/python/nn
i-robot eac31a1a40
!44008 Add nn.GaussianNLLLoss
Merge pull request !44008 from shaojunsong/feature/nnBilinear
2022-10-19 06:47:38 +00:00
..
gradient modify vjp 2022-09-14 15:08:16 +08:00
layer fractionalmaxpool 2022-10-17 15:20:54 +08:00
optim Rectify Adam Optimizer 2022-09-26 22:17:19 +08:00
probability modify ut of probability 2021-10-14 19:33:34 +08:00
__init__.py pylint waring clean 2020-05-13 11:30:27 +08:00
test_activation.py Add nn.Softmax2d 2022-10-10 21:22:48 +08:00
test_batchnorm.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_cell.py Fix fallback and cell kwargs issues. 2021-10-27 14:06:02 +08:00
test_cell_method_attribute.py bug0 2022-10-18 11:56:08 +08:00
test_cell_wrapper.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_central_crop.py Optimize the Executors routines. 2021-08-28 17:36:12 +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 Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_dynamic_lr.py uniform learning_rate behavior of optimizers 2020-07-20 21:09:20 +08:00
test_flatten.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_l1_regularizer.py add security isolate for save_graphs 2021-09-14 10:24:12 +08:00
test_learning_rate_schedule.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_loss.py Add nn.GaussianNLLLoss 2022-10-18 20:07:06 +08:00
test_lstm.py pylint waring clean 2020-05-13 11:30:27 +08:00
test_map_parameter.py Implement backward graph for MapParameter 2022-10-17 17:37:01 +08:00
test_msssim.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_nn_embedding.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_nn_pad.py remove name arg from gradoperation 2020-08-26 11:16:57 +08:00
test_nn_padding.py add nn.ConstantPad1d nn.ConstantPad2d nn.ConstantPad3d and nn.ZeroPad2d 2022-05-12 10:34:52 +08:00
test_nn_reflectionpad.py added reflectionpad1d and 2d 2022-06-04 17:43:49 +08:00
test_norm.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_parameter.py delete auto parallel flag 2022-07-30 15:19:56 +08:00
test_pooling.py [feat] [assistant] [ops] [I51VS2] Add new operator nn.AdaptiveMaxPool1d 2022-05-18 15:50:44 +08:00
test_psnr.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_resizebilinear.py Modify error content 2021-09-10 15:21:33 +08:00
test_ssim.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_transformer.py Fix activation shard error 2022-10-08 20:39:55 +08:00
test_tril.py Add nn.Tril function 2020-11-19 10:31:30 +08:00
test_triu.py "change default backend in debug graph mode and set uni folder for ge 2022-05-20 09:12:03 +08:00