| .. |
|
addn_gpu_kernel.cc
|
add float64 to Addn gpu
|
2021-03-19 15:46:27 -04:00 |
|
addn_gpu_kernel.h
|
add float64 to Addn gpu
|
2021-03-19 15:46:27 -04:00 |
|
assign_add_gpu_kernel.cc
|
fix isnan isfinite isinfinite infer type
|
2020-12-02 20:19:15 +08:00 |
|
assign_add_gpu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
broadcast_gpu_kernel.cc
|
!12897 Add float64 support to reducemax grad gpu op
|
2021-03-08 19:41:35 +08:00 |
|
broadcast_gpu_kernel.h
|
Change TensorAdd to Add, from r1.1 to master
|
2021-02-01 17:53:52 +08:00 |
|
broadcast_grad_gpu_kernel.cc
|
fix isnan isfinite isinfinite infer type
|
2020-12-02 20:19:15 +08:00 |
|
broadcast_grad_gpu_kernel.h
|
add trace for gpu error/excpt log
|
2020-12-04 14:03:36 +08:00 |
|
cast_all_gpu_kernel.cc
|
gpu add combine cast fusion
|
2020-09-27 16:07:07 +08:00 |
|
cast_all_gpu_kernel.h
|
add trace for gpu error/excpt log
|
2020-12-04 14:03:36 +08:00 |
|
cholesky_solve_gpu_kernel.cc
|
Add Cholesky op at GPU back-end
|
2020-11-30 20:27:22 -05:00 |
|
cholesky_solve_gpu_kernel.h
|
add trace for gpu error/excpt log
|
2020-12-04 14:03:36 +08:00 |
|
cholesky_trsm_solve_gpu_kernel.cc
|
Refactor Cholesky to CholeskyTrsm
|
2020-11-03 14:56:30 -05:00 |
|
cholesky_trsm_solve_gpu_kernel.h
|
add trace for gpu error/excpt log
|
2020-12-04 14:03:36 +08:00 |
|
cumsum_gpu_kernel.cc
|
initial commit
|
2021-02-17 17:39:18 -05:00 |
|
cumsum_gpu_kernel.h
|
support int64 shape
|
2020-11-05 21:15:49 +08:00 |
|
determinant_triangle_gpu_kernel.cc
|
Add calculation of triangle matrix determinant op at GPU back-end
|
2020-10-27 15:20:23 -04:00 |
|
determinant_triangle_gpu_kernel.h
|
support int64 shape
|
2020-11-05 21:15:49 +08:00 |
|
equalcount_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
equalcount_gpu_kernel.h
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
float_status_gpu_kernel.cc
|
change float status output type to float32
|
2021-02-18 16:16:26 +08:00 |
|
float_status_gpu_kernel.h
|
change float status output type to float32
|
2021-02-18 16:16:26 +08:00 |
|
index_add_gpu_kernel.cc
|
index_add gpu op
|
2021-02-19 09:49:06 -05:00 |
|
index_add_gpu_kernel.h
|
index_add gpu op
|
2021-02-19 09:49:06 -05:00 |
|
linspace.cc
|
Initial Commit - GPU LinSpace
|
2020-11-26 02:33:55 -05:00 |
|
linspace.h
|
Initial Commit - GPU LinSpace
|
2020-11-26 02:33:55 -05:00 |
|
matmul_gpu_kernel.cc
|
add float64 support to matmul ops
|
2021-02-17 12:35:55 -05:00 |
|
matmul_gpu_kernel.h
|
add float64 support to matmul ops
|
2021-02-17 12:35:55 -05:00 |
|
matrix_inverse_gpu_kernel.cc
|
new add matrix inverse.
|
2020-12-29 16:34:21 +08:00 |
|
matrix_inverse_gpu_kernel.h
|
new add matrix inverse.
|
2020-12-29 16:34:21 +08:00 |
|
multinomial_gpu_kernel.cc
|
add multinomial backend
|
2020-08-11 17:18:59 +08:00 |
|
multinomial_gpu_kernel.h
|
support int64 shape
|
2020-11-05 21:15:49 +08:00 |
|
nms_with_mask_gpu_kernel.cc
|
stuff added
|
2020-07-29 15:11:24 -04:00 |
|
nms_with_mask_gpu_kernel.h
|
NMS perf boost
|
2020-09-15 18:01:45 -04:00 |
|
random_op_gpu_kernel.cc
|
substitute dropout by cudnnuniformreal and dropout
|
2020-11-26 10:36:02 +08:00 |
|
random_op_gpu_kernel.h
|
Fix an error that uniform_a should be less than uniform_b
|
2020-11-30 20:04:37 -05:00 |
|
square_sum_all_gpu_kernel.cc
|
support squaresumall on gpu
|
2020-12-28 09:11:57 +08:00 |
|
square_sum_all_gpu_kernel.h
|
fix a bug for squaresumall in fp16
|
2021-02-26 14:19:05 +08:00 |
|
squared_difference_kernel.cc
|
Add SquareDifference kernel for GPU
|
2020-12-28 16:47:52 +08:00 |
|
squared_difference_kernel.h
|
Add SquareDifference kernel for GPU
|
2020-12-28 16:47:52 +08:00 |
|
unary_op_gpu_kernel.cc
|
removed zeroslike from math/unary_op_gpu_kernel and added float64 support
|
2021-02-22 16:15:23 -05:00 |
|
unary_op_gpu_kernel.h
|
removed zeroslike from math/unary_op_gpu_kernel and added float64 support
|
2021-02-22 16:15:23 -05:00 |
|
unary_op_grad_gpu_kernel.cc
|
add float64 support to absgrad and sqrtgrad
|
2021-03-03 17:18:00 -05:00 |
|
unary_op_grad_gpu_kernel.h
|
add new GPU kernel ReciprocalGrad
|
2021-01-27 17:48:01 -05:00 |
|
update_thor_gradient.cc
|
update thor gradient op
|
2020-08-05 17:37:08 +08:00 |
|
update_thor_gradient.h
|
add trace for gpu error/excpt log
|
2020-12-04 14:03:36 +08:00 |
b780e5737c