You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

tensor-tang 3d950a812d combine jitcode of vscal		7 years ago
..
detail	Refine activation for GRU operator (#13275)	7 years ago
CMakeLists.txt	fix jit on mac	7 years ago
algorithm.h	test=develop	7 years ago
blas.cc	…
blas.h	…
blas_impl.cu.h	…
blas_impl.h	…
compound_functors.h	Feature/op_fuse_pass (#12440)	7 years ago
concat.hip.cu	…
concat_and_split.cc	Refine Split op (#13967)	7 years ago
concat_and_split.cu	Refine Split op (#13967)	7 years ago
concat_and_split.h	Refine Split op (#13967)	7 years ago
concat_test.cc	Refine Split op (#13967)	7 years ago
context_project.cc	…
context_project.cu	…
context_project.h	…
cos_sim_functor.cc	…
cos_sim_functor.cu	Fix SelectedRowsAdd bug (#14309)	7 years ago
cos_sim_functor.h	…
cpu_vec.h	refine vmul	7 years ago
cpu_vec_test.cc	refine vmul	7 years ago
cross_entropy.cc	…
cross_entropy.cu	hot fix log (#14293)	7 years ago
cross_entropy.h	Add fp16 backward support (#14202)	7 years ago
depthwise_conv.cu	optimize depthwise conv by register memory (#13778)	7 years ago
depthwise_conv.h	Optimization of Kernels that related to DeepLabv3+ (#13534)	7 years ago
fc_compute.h	fix compile error	7 years ago
functors.h	Feature/op_fuse_pass (#12440)	7 years ago
gru_compute.cc	…
gru_compute.cu	…
gru_compute.h	…
im2col.cc	…
im2col.cu	…
im2col.h	…
im2col_cfo_cpu.h	…
im2col_test.cc	…
jit_code.cc	combine jitcode of vscal	7 years ago
jit_code.h	combine jitcode of vscal	7 years ago
jit_gen.cc	add jit gencode	7 years ago
jit_gen.h	add jit gencode	7 years ago
jit_kernel.cc	thread local jit kernels	7 years ago
jit_kernel.h	add vscal jitcode	7 years ago
jit_kernel_blas.cc	combine jitcode of vscal	7 years ago
jit_kernel_crf_decode.cc	refine and init jitkernel vmul	7 years ago
jit_kernel_exp.cc	add vscal jitcode	7 years ago
jit_kernel_macro.h	refine and init jitkernel vmul	7 years ago
jit_kernel_rnn.cc	add vadd jitcode	7 years ago
jit_kernel_test.cc	add vscal jitcode	7 years ago
lstm_compute.cc	…
lstm_compute.cu	…
lstm_compute.h	…
math_function.cc	"fix compile error" (#13579)	7 years ago
math_function.cu	…
math_function.h	"fix compile error" (#13579)	7 years ago
math_function_impl.h	…
math_function_test.cc	…
math_function_test.cu	…
matrix_bit_code.cc	…
matrix_bit_code.h	…
maxouting.cc	…
maxouting.cu	…
maxouting.h	…
padding.h	…
pooling.cc	move param exclusive to the last in pool2d/pool3d for forward compatibility:. test=develop	7 years ago
pooling.cu	move param exclusive to the last in pool2d/pool3d for forward compatibility:. test=develop	7 years ago
pooling.h	add inclusive/exclusive mode in PoolOp avg pool type	7 years ago
sampler.cc	…
sampler.h	…
selected_rows_functor.cc	optimize code	7 years ago
selected_rows_functor.cu	Fix SelectedRowsAdd bug (#14309)	7 years ago
selected_rows_functor.h	delete buggy selected_rows functor	7 years ago
selected_rows_functor_test.cc	sum op handle empty input	7 years ago
selected_rows_functor_test.cu	sum op handle empty input	7 years ago
sequence2batch.cc	…
sequence2batch.cu	…
sequence2batch.h	…
sequence_padding.cc	…
sequence_padding.cu	…
sequence_padding.h	…
sequence_padding_test.cc	…
sequence_pooling.cc	Rebase of seqpool-max optimization	7 years ago
sequence_pooling.cu	Rebase of seqpool-max optimization	7 years ago
sequence_pooling.h	Rebase of seqpool-max optimization	7 years ago
sequence_pooling_test.cc	Fix copy wrong pos bug	7 years ago
sequence_scale.cc	…
sequence_scale.cu	…
sequence_scale.h	…
softmax.cc	…
softmax.cu	Add fp16 backward support (#14202)	7 years ago
softmax.h	…
softmax_impl.h	…
unpooling.cc	…
unpooling.cu	…
unpooling.h	…
vol2col.cc	…
vol2col.cu	…
vol2col.h	…
vol2col_test.cc	…