You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
mindspore/tests/ut/python/nn
buxue 4168c51f0d
add check tensor when graph output is a const or parameter
5 years ago
..
bijector add scalar affine and softplus bijector 5 years ago
distribution fix bernoulli prob formula; fix some other minor bugs 5 years ago
optim remove name arg from gradoperation 5 years ago
probability add infer and dpn 5 years ago
__init__.py pylint waring clean 5 years ago
test_activation.py pylint waring clean 5 years ago
test_batchnorm.py pylint waring clean 5 years ago
test_cell.py pylint waring clean 5 years ago
test_cell_wrapper.py implicit type conversion 5 years ago
test_central_crop.py Add image.CentralCrop 5 years ago
test_checkparameter.py pylint waring clean 5 years ago
test_clip_by_norm.py clean pylint 5 years ago
test_container.py pylint clean 5 years ago
test_conv.py !906 fix a bug that support dilation greater than 1 in conv2dTranspose ops 5 years ago
test_dense.py fix the condition when activation name is 0 5 years ago
test_dropout.py pylint waring clean 5 years ago
test_dynamic_lr.py uniform learning_rate behavior of optimizers 5 years ago
test_flatten.py pylint waring clean 5 years ago
test_image_gradients.py pylint waring clean 5 years ago
test_learning_rate_schedule.py uniform learning_rate behavior of optimizers 5 years ago
test_loss.py composed op CosineEmbeddingLoss 5 years ago
test_lstm.py pylint waring clean 5 years ago
test_msssim.py update SSIM loss, add MSSSIM loss feature; add their ut testcases. 5 years ago
test_nn_embedding.py pylint waring clean 5 years ago
test_nn_pad.py remove name arg from gradoperation 5 years ago
test_norm.py pylint waring clean 5 years ago
test_parameter.py fix bug in Parameter init_data 5 years ago
test_pooling.py clean pylint 5 years ago
test_psnr.py clean pylint 5 years ago
test_ssim.py update SSIM loss, add MSSSIM loss feature; add their ut testcases. 5 years ago