| .. |
|
detail
|
Renamed to strided_memcpy and prettify unittests
|
8 years ago |
|
math
|
fix SoftmaxWithCrossEntropyOp
|
8 years ago |
|
rnn
|
clean code
|
8 years ago |
|
.clang-format
|
Unify clang-format and add some missing clang-format
|
8 years ago |
|
CMakeLists.txt
|
cross entropy as a functor to avoid duplicated codes.
|
8 years ago |
|
accuracy_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
accuracy_op.cu
|
fix cpu kernel with soft labels.
|
8 years ago |
|
accuracy_op.h
|
Accuracy op (#3907)
|
8 years ago |
|
activation_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
activation_op.cu
|
fix gpu build error
|
8 years ago |
|
activation_op.h
|
add activation operators and python unittests
|
8 years ago |
|
add_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
add_op.cu
|
…
|
|
|
add_op.h
|
…
|
|
|
clip_op.cc
|
Fix comments
|
8 years ago |
|
clip_op.cu
|
Update transform invocation
|
8 years ago |
|
clip_op.h
|
Fix ptr type
|
8 years ago |
|
concat_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
concat_op.h
|
…
|
|
|
cond_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
cond_op.h
|
remove empty line
|
8 years ago |
|
conv2d_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
conv2d_op.cu
|
Some bug fix.
|
8 years ago |
|
cos_sim_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
cos_sim_op.cu
|
…
|
|
|
cos_sim_op.h
|
Fix Clang compile error
|
8 years ago |
|
crop_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
crop_op.cu
|
Use stridecpy instead of CUDA kernel
|
8 years ago |
|
crop_op.h
|
add namespace and reduce warning
|
8 years ago |
|
cross_entropy_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
cross_entropy_op.cu
|
cross entropy as a functor to avoid duplicated codes.
|
8 years ago |
|
cross_entropy_op.h
|
cross entropy as a functor to avoid duplicated codes.
|
8 years ago |
|
dropout_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
dropout_op.cu
|
Add bool type for attribute and use it in dropout_op.
|
8 years ago |
|
dropout_op.h
|
Add bool type for attribute and use it in dropout_op.
|
8 years ago |
|
elementwise_add_op.cc
|
split elementwise_op.h into two header files
|
8 years ago |
|
elementwise_add_op.cu
|
Elementwise operator. (#4139)
|
8 years ago |
|
elementwise_add_op.h
|
split elementwise_op.h into two header files
|
8 years ago |
|
elementwise_div_op.cc
|
split elementwise_op.h into two header files
|
8 years ago |
|
elementwise_div_op.cu
|
Elementwise operator. (#4139)
|
8 years ago |
|
elementwise_div_op.h
|
split elementwise_op.h into two header files
|
8 years ago |
|
elementwise_mul_op.cc
|
split elementwise_op.h into two header files
|
8 years ago |
|
elementwise_mul_op.cu
|
Elementwise operator. (#4139)
|
8 years ago |
|
elementwise_mul_op.h
|
split elementwise_op.h into two header files
|
8 years ago |
|
elementwise_op.h
|
add include op_registry.h to elementwise_op_function.h
|
8 years ago |
|
elementwise_op_function.h
|
add include op_registry.h to elementwise_op_function.h
|
8 years ago |
|
elementwise_sub_op.cc
|
split elementwise_op.h into two header files
|
8 years ago |
|
elementwise_sub_op.cu
|
Elementwise operator. (#4139)
|
8 years ago |
|
elementwise_sub_op.h
|
split elementwise_op.h into two header files
|
8 years ago |
|
fc_op.cc
|
Fix bug
|
8 years ago |
|
fill_zeros_like_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
fill_zeros_like_op.cu
|
…
|
|
|
fill_zeros_like_op.h
|
Share LoD between input and output of each opeators.
|
8 years ago |
|
gather.h
|
…
|
|
|
gather_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
gather_op.h
|
…
|
|
|
gather_test.cc
|
…
|
|
|
gaussian_random_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
gaussian_random_op.cu
|
…
|
|
|
gemm_conv2d_op.h
|
Refine code.
|
8 years ago |
|
identity_op.cc
|
Delete USE_OP statements and add more ENFORCE statements to check the inputs and outputs in FCOp.
|
8 years ago |
|
lookup_table_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
lookup_table_op.cu
|
fix cpu kernel with soft labels.
|
8 years ago |
|
lookup_table_op.h
|
…
|
|
|
lstm_unit_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
lstm_unit_op.cu
|
lstm unit gpu
|
8 years ago |
|
lstm_unit_op.h
|
lstm
|
8 years ago |
|
mean_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
mean_op.cu
|
…
|
|
|
mean_op.h
|
…
|
|
|
minus_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
minus_op.cu
|
…
|
|
|
minus_op.h
|
…
|
|
|
modified_huber_loss_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
modified_huber_loss_op.cu
|
Fix some conflicts and correct unittest.
|
8 years ago |
|
modified_huber_loss_op.h
|
update to develop branch.
|
8 years ago |
|
mul_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
mul_op.cu
|
…
|
|
|
mul_op.h
|
Remove lazy-initialization in device_context
|
8 years ago |
|
multiplex_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
multiplex_op.cu
|
fix compiler warning from MKLDNNLayer and so on
|
8 years ago |
|
multiplex_op.h
|
apply more general dims for multiplex_op
|
8 years ago |
|
name_convention.md
|
"better format"
|
8 years ago |
|
net_op.cc
|
…
|
|
|
net_op.h
|
…
|
|
|
net_op_design.md
|
…
|
|
|
net_op_test.cc
|
…
|
|
|
pad_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
pad_op.cu
|
…
|
|
|
pad_op.h
|
Fix nullptr check
|
8 years ago |
|
prelu_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
prelu_op.cu
|
prelu with gpu
|
8 years ago |
|
prelu_op.h
|
remove alias
|
8 years ago |
|
rank_loss_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
rank_loss_op.cu
|
add rank_loss operator
|
8 years ago |
|
rank_loss_op.h
|
update to develop branch.
|
8 years ago |
|
recurrent_op.cc
|
make RecurrentOp's backward work
|
8 years ago |
|
recurrent_op.h
|
make RecurrentOp's backward work
|
8 years ago |
|
reshape_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
reshape_op.cu
|
…
|
|
|
reshape_op.h
|
fix cpplint error
|
8 years ago |
|
rnn_design.md
|
…
|
|
|
rowwise_add_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
rowwise_add_op.cu
|
…
|
|
|
rowwise_add_op.h
|
…
|
|
|
scale_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
scale_op.cu
|
…
|
|
|
scale_op.h
|
…
|
|
|
scatter.h
|
…
|
|
|
scatter_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
scatter_op.h
|
…
|
|
|
scatter_test.cc
|
…
|
|
|
sequence_pool_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
sequence_pool_op.cu
|
implement framework of seq_pool_op and its unitest
|
8 years ago |
|
sequence_pool_op.h
|
Refactoring InferShape (#3946)
|
8 years ago |
|
sgd_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
sgd_op.cu
|
…
|
|
|
sgd_op.h
|
…
|
|
|
smooth_l1_loss_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
smooth_l1_loss_op.cu
|
…
|
|
|
smooth_l1_loss_op.h
|
Refine doc.
|
8 years ago |
|
softmax_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
softmax_op.cu
|
…
|
|
|
softmax_op.h
|
fix implementations of supporting soft labels.
|
8 years ago |
|
softmax_with_cross_entropy_op.cc
|
fix SoftmaxWithCrossEntropyOp
|
8 years ago |
|
softmax_with_cross_entropy_op.cu
|
cross entropy as a functor to avoid duplicated codes.
|
8 years ago |
|
softmax_with_cross_entropy_op.h
|
cross entropy as a functor to avoid duplicated codes.
|
8 years ago |
|
split_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
split_op.h
|
Split operator with CPU kernel (#4046)
|
8 years ago |
|
squared_l2_distance_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
squared_l2_distance_op.cu
|
…
|
|
|
squared_l2_distance_op.h
|
…
|
|
|
strided_memcpy.h
|
Renamed to strided_memcpy and prettify unittests
|
8 years ago |
|
strided_memcpy_test.cc
|
Renamed to strided_memcpy and prettify unittests
|
8 years ago |
|
sum_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
sum_op.cu
|
…
|
|
|
sum_op.h
|
…
|
|
|
top_k_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
top_k_op.cu
|
fix cpu kernel with soft labels.
|
8 years ago |
|
top_k_op.h
|
…
|
|
|
transpose_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
transpose_op.cu
|
delete cuda impl, complete comments, modify variable naming
|
8 years ago |
|
transpose_op.h
|
modify the input\output name to X\Out
|
8 years ago |
|
uniform_random_op.cc
|
Refactoring InferShape (#3946)
|
8 years ago |
|
uniform_random_op.cu
|
…
|
|
Yu Yang