onnx-jax:基于jax的onnx运行时
需积分: 10 74 浏览量
2021-04-18
00:43:16
上传
评论
收藏 94KB ZIP 举报
onnx-jax:基于jax的onnx运行时 （170个子文件） 
.env 44B .gitignore 19B demo.ipynb 16KB launch.json 659B settings.json 232B README.md 219B op_states.md 217B test_averagepool.py 14KB test_argmin.py 9KB test_argmax.py 8KB test_conv.py 8KB test_gemm.py 6KB test_cast.py 5KB backend.py 5KB test_clip.py 4KB conv_transpose.py 4KB test_bitshift.py 4KB average_pool.py 3KB test_compress.py 3KB test_log_softmax.py 3KB slice.py 2KB test_and.py 2KB conv.py 2KB maxpool.py 2KB test_batchnorm.py 2KB test_max.py 2KB concat.py 2KB cast.py 2KB gemm.py 2KB test_flatten.py 2KB test_instancenorm.py 2KB argmin.py 2KB argmax.py 2KB test_concat.py 2KB handler.py 2KB add.py 2KB demo.py 2KB pb_wrapper.py 2KB batchnorm.py 2KB handler_helper.py 2KB test_identity.py 1KB compress.py 1KB clip.py 1KB test_celu.py 1KB flatten.py 1KB max.py 1KB instancenorm.py 1KB reshape.py 1KB test_global_average_pool.py 1KB test_global_max_pool.py 1KB pad.py 1KB dropout.py 1KB log_softmax.py 1KB split.py 1KB constant.py 1KB test_leakyrelu.py 1KB test_hardsigmoid.py 1KB hardswish.py 1KB test_elu.py 1KB hardsigmoid.py 1KB test_div.py 1KB upsample.py 1KB elu.py 1KB leaky_relu.py 1KB test_greater_or_equal.py 1KB test_less_equal.py 1KB test_greater.py 994B bitshift.py 991B test_matmul.py 990B mean.py 986B test_equal.py 980B reduce_max.py 959B reduce_min.py 959B test_less.py 955B test_det.py 918B reduce_sum.py 916B mul.py 891B sum.py 885B tools.py 882B test_add.py 869B min.py 869B gather.py 868B celu.py 856B reduce_mean.py 852B gather_elements.py 836B test_acosh.py 824B greater.py 819B reduce_log_sum_exp.py 816B equal.py 810B reduce_sum_square.py 808B less.py 804B div.py 802B reduce_log_sum.py 793B reduce_l2.py 790B p_relu.py 778B reduce_l1.py 777B reduce_prod.py 775B sub.py 756B pow.py 754B test_acos.py 752B共 170 条
- 1
- 2
资源评论
