| .. |
|
gradient
|
add examples of grad, jvp and vjp
|
2021-11-16 14:15:26 +08:00 |
|
optim
|
Support fusion_size argument for flatten_weights
|
2022-05-20 17:11:48 +08:00 |
|
probability
|
modify ut of probability
|
2021-10-14 19:33:34 +08:00 |
|
__init__.py
|
…
|
|
|
test_activation.py
|
!33877 operators: Softmin, SiLU
|
2022-06-01 01:53:19 +00: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_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
|
…
|
|
|
test_clip_by_norm.py
|
…
|
|
|
test_container.py
|
set default context mode to GRAPH_MODE
|
2021-06-09 14:40:49 +08:00 |
|
test_conv.py
|
…
|
|
|
test_dense.py
|
Optimize the Executors routines.
|
2021-08-28 17:36:12 +08:00 |
|
test_dynamic_lr.py
|
…
|
|
|
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
|
…
|
|
|
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
|
…
|
|
|
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
|
Modify Optimizer to support flatten parameters
|
2022-04-07 14:11:32 +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
|
Add args check
|
2021-09-08 09:12:35 +08:00 |
|
test_tril.py
|
…
|
|
|
test_triu.py
|
"change default backend in debug graph mode and set uni folder for ge
|
2022-05-20 09:12:03 +08:00 |
i-robot
f058997021