mindspore/tests/ut/python/nn
huangxinjing 0e575f178d Fix activation shard error
Fix hidden activation error
2022-10-08 20:39:55 +08:00
..
gradient modify vjp 2022-09-14 15:08:16 +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 delete extra check 2022-06-30 18:29:37 +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 Support to use methods/attributes of Cell object in graph compiler. 2022-07-07 16:18:25 +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 nll_loss and cross_entropy for ops.functional and nn 2022-06-02 11:13:34 +08:00
test_lstm.py pylint waring clean 2020-05-13 11:30:27 +08:00
test_map_parameter.py Implement MapParameter default value and clone() 2022-09-29 08:50:30 +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