|
_grad
|
fix sparse feature bug for auto parallel
|
2020-07-25 17:26:43 +08:00 |
|
_op_impl
|
Enable get rank id and size by group
|
2020-07-21 23:49:52 +08:00 |
|
_utils
|
support tensor get value by tensor index
|
2020-05-21 21:05:35 +08:00 |
|
composite
|
Fix minor errors in probabilistic programming
|
2020-07-28 11:35:10 -04:00 |
|
operations
|
add maxpool_with_argmax/grad cuda kernel
|
2020-07-30 23:38:35 +08:00 |
|
__init__.py
|
Refactor the akg op registers.
|
2020-07-14 15:06:12 +08:00 |
|
_register_for_op.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
_selected_grad_ops.py
|
Init GraphKernel.
|
2020-06-20 22:31:54 +08:00 |
|
_selected_ops.py
|
Init GraphKernel.
|
2020-06-20 22:31:54 +08:00 |
|
op_info_register.py
|
Refactor the akg op registers.
|
2020-07-14 15:06:12 +08:00 |
|
op_selector.py
|
Init GraphKernel.
|
2020-06-20 22:31:54 +08:00 |
|
vm_impl_registry.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
panfengfeng
ca881ec03e