|
_grad
|
!3746 add klDivLoss python primitive
|
2020-07-31 14:20:13 +08:00 |
|
_op_impl
|
!3547 fix serving core dump problem
|
2020-07-28 20:22:39 +08:00 |
|
_utils
|
support tensor get value by tensor index
|
2020-05-21 21:05:35 +08:00 |
|
composite
|
add sparse operators
|
2020-07-30 17:18:23 +08:00 |
|
operations
|
!3746 add klDivLoss python primitive
|
2020-07-31 14:20:13 +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
|
Add 13 akg/ascend op registers
|
2020-07-22 14:14:32 +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 |