| .. |
|
__init__.py
|
Adam op performance optimization
|
2021-08-27 14:58:07 +08:00 |
|
adam.py
|
add fp16 support for CPU Adam
|
2021-05-28 16:52:08 +08:00 |
|
adam_weight_decay.py
|
Adam op performance optimization
|
2021-08-27 14:58:07 +08:00 |
|
add.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
arg_max.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
arg_max_with_value.py
|
adding 9 object detection operators
|
2021-07-21 10:49:17 +08:00 |
|
arg_min_with_value.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
bias_add.py
|
revert reimplementation of biasAdd using nnacl
|
2021-06-15 16:59:23 +08:00 |
|
bias_add_grad.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
buffer_append.py
|
add prim for replay buffer
|
2021-08-11 09:19:42 +08:00 |
|
buffer_get.py
|
add prim for replay buffer
|
2021-08-11 09:19:42 +08:00 |
|
buffer_sample.py
|
add prim for replay buffer
|
2021-08-11 09:19:42 +08:00 |
|
cast.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
concat.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
conv2d.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
conv3d.py
|
fix conv3d when pad_mode is pad
|
2021-07-06 17:06:23 +08:00 |
|
div.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
dropout.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
dropout_grad.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
equal_count.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
gather_d.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
gather_d_grad.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
gather_nd.py
|
complete forward and backward of SparseToDense and check of SparseTensorDenseMatmul
|
2021-06-01 17:40:45 +08:00 |
|
gather_v2.py
|
complete forward and backward of SparseToDense and check of SparseTensorDenseMatmul
|
2021-06-01 17:40:45 +08:00 |
|
hsigmoid.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
hsigmoid_grad.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
hswish.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
hswish_grad.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
is_finite.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
l2loss.py
|
add CPU l2loss op
|
2021-08-03 20:19:15 +08:00 |
|
layer_norm.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
layer_norm_grad.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
maximum.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
maximum_grad.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
minimum.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
minimum_grad.py
|
fix code review problems
|
2021-06-15 10:54:22 +08:00 |
|
mirror_pad.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
mirror_pad_grad.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
mul.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
one_hot.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
pad.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
pow.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
pyfunc.py
|
pyfunc primitive register
|
2021-08-10 15:47:07 +08:00 |
|
range.py
|
pclint clean
|
2021-06-02 09:52:32 +08:00 |
|
real_div.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
reduce_all.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
reduce_any.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
reduce_max.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
reduce_mean.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
reduce_min.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
reduce_sum.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
split.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
stack.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
sub.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
tensor_copy_slices.py
|
Add TensorCopySlice Op
|
2021-06-29 16:52:35 +08:00 |
|
tile.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
top_k.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
|
transpose.py
|
upload unify cpu ops info library code
|
2021-05-26 16:29:25 +08:00 |
zhaosida
be34ccd29f