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.

chengduo a7497653d0 Refine Split op (#13967) ... * speedup split_op test=develop * speedup split_op test=develop * rename ConcatGrad to Split * refine concat and split test=develop * fix compile error		6 years ago
..
detail	Refine activation for GRU operator (#13275)	7 years ago
CMakeLists.txt	Refine Split op (#13967)	6 years ago
algorithm.h	fix sparse rmsprop	6 years ago
blas.cc	…
blas.h	fix bugs	7 years ago
blas_impl.cu.h	…
blas_impl.h	fix bugs	7 years ago
compound_functors.h	Feature/op_fuse_pass (#12440)	7 years ago
concat.hip.cu	…
concat_and_split.cc	Refine Split op (#13967)	6 years ago
concat_and_split.cu	Refine Split op (#13967)	6 years ago
concat_and_split.h	Refine Split op (#13967)	6 years ago
concat_test.cc	Refine Split op (#13967)	6 years ago
context_project.cc	…
context_project.cu	…
context_project.h	…
cos_sim_functor.cc	…
cos_sim_functor.cu	…
cos_sim_functor.h	…
cpu_vec.h	refine vmul	6 years ago
cpu_vec_test.cc	refine vmul	6 years ago
cross_entropy.cc	Add ignore_index in cross_entropy op (#13217)	7 years ago
cross_entropy.cu	Add ignore_index in cross_entropy op (#13217)	7 years ago
cross_entropy.h	Add ignore_index in cross_entropy op (#13217)	7 years ago
depthwise_conv.cu	optimize depthwise conv by register memory (#13778)	6 years ago
depthwise_conv.h	Optimization of Kernels that related to DeepLabv3+ (#13534)	6 years ago
fc_compute.h	refine and add eltadd_relu unit test	6 years ago
functors.h	Feature/op_fuse_pass (#12440)	7 years ago
gru_compute.cc	…
gru_compute.cu	…
gru_compute.h	…
im2col.cc	complete im2col with padding==1 and speedup filter width==1	7 years ago
im2col.cu	change im2sequence for ctc batch inference (#11696)	7 years ago
im2col.h	…
im2col_cfo_cpu.h	further optimize im2col using variables	7 years ago
im2col_test.cc	complete im2col with padding==1 and speedup filter width==1	7 years ago
jit_kernel.cc	thread local jit kernels	6 years ago
jit_kernel.h	fuse elementwise_add and relu	6 years ago
jit_kernel_blas.cc	refine and add eltadd_relu unit test	6 years ago
jit_kernel_exp.cc	clean code exp avx	6 years ago
jit_kernel_lstm.cc	fix illegal instruction of rnn2	6 years ago
jit_kernel_macro.h	refine code and fix	6 years ago
jit_kernel_test.cc	refine and add eltadd_relu unit test	6 years ago
lstm_compute.cc	…
lstm_compute.cu	…
lstm_compute.h	…
math_function.cc	"fix compile error" (#13579)	6 years ago
math_function.cu	Feature/template (#13093)	7 years ago
math_function.h	"fix compile error" (#13579)	6 years ago
math_function_impl.h	…
math_function_test.cc	add gemm_warp test	7 years ago
math_function_test.cu	…
matrix_bit_code.cc	Fix the HierarchicalSigmoidGradOpKernel and refine the codes. Now hsigmoid_op is same with V2 implementation and can pass gradient check.	7 years ago
matrix_bit_code.h	fix style (#13142)	7 years ago
maxouting.cc	…
maxouting.cu	…
maxouting.h	fix style (#13142)	7 years ago
padding.h	Add pad_constant_like_op (#12943)	7 years ago
pooling.cc	…
pooling.cu	…
pooling.h	fix style (#13142)	7 years ago
sampler.cc	…
sampler.h	…
selected_rows_functor.cc	test=develop	6 years ago
selected_rows_functor.cu	Merge pull request #13525 from reyoung/fix_mixed_vector	6 years ago
selected_rows_functor.h	test=develop	6 years ago
selected_rows_functor_test.cc	Accelerate SelectedRows Functors:	6 years ago
selected_rows_functor_test.cu	Fix unstable selected_rows_functor_test.cu	7 years ago
sequence2batch.cc	refine seq2batch	7 years ago
sequence2batch.cu	…
sequence2batch.h	refine cpu gru batch mode	7 years ago
sequence_padding.cc	update function	7 years ago
sequence_padding.cu	update function	7 years ago
sequence_padding.h	bug fix	7 years ago
sequence_padding_test.cc	Complete sequence_padding GPU kernel	7 years ago
sequence_pooling.cc	Accelerate SequencePool Op on SUM mode	6 years ago
sequence_pooling.cu	refine seq_pool	7 years ago
sequence_pooling.h	…
sequence_scale.cc	…
sequence_scale.cu	…
sequence_scale.h	…
softmax.cc	…
softmax.cu	Revert ""cherry picked operators changes" (#12184)" (#12747)	7 years ago
softmax.h	…
softmax_impl.h	…
unpooling.cc	…
unpooling.cu	…
unpooling.h	…
vol2col.cc	…
vol2col.cu	…
vol2col.h	…
vol2col_test.cc	…