mindspore/tests/ut/python/nn
pkuliuliu e0f470751b add nn.ChannelShuffle 2022-10-31 21:15:12 +08:00
..
gradient modify vjp 2022-09-14 15:08:16 +08:00
layer add nn.ChannelShuffle 2022-10-31 21:15:12 +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_hinge_embedding_loss.py Add nn.HingeEmbeddingLoss 2022-10-24 17:30:41 +08:00
test_l1_regularizer.py Add api mindspore.jit and mindspore.jit_class. 2022-10-21 10:17:44 +08:00
test_learning_rate_schedule.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_loss.py add ctcloss 2022-10-24 09:32:19 +08:00
test_lstm.py pylint waring clean 2020-05-13 11:30:27 +08:00
test_map_parameter.py Support new MapTensor in construct. 2022-10-28 19:51:27 +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 Add api mindspore.jit and mindspore.jit_class. 2022-10-21 10:17:44 +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 add tensor dim(), numpy(), param copy 2022-10-19 23:46:08 +08:00
test_pooling.py add function api of max_unpool1d, 2d, 3d 2022-10-27 09:42:46 +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 [AutoParallel]Fix batch infer error 2022-10-11 11:59:53 +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