cavis/libnd4j/include/ops/declarable/generic/nn
agibsonccc 8bc3172e40 Fix NCHW case for fused batch norm 2021-02-16 11:02:27 +09:00
..
activations Update c++ copyrights 2021-02-01 21:31:45 +09:00
convo Update copyrights remove attic and relocate elsewhere 2021-02-09 13:16:31 +09:00
pooling Update c++ copyrights 2021-02-01 21:31:45 +09:00
recurrent Update copyrights remove attic and relocate elsewhere 2021-02-09 13:16:31 +09:00
apply_sgd.cpp Update c++ copyrights 2021-02-01 21:31:45 +09:00
batchnorm.cpp Update copyrights remove attic and relocate elsewhere 2021-02-09 13:16:31 +09:00
bias_add.cpp Update c++ copyrights 2021-02-01 21:31:45 +09:00
dot_product_attention.cpp Update copyrights remove attic and relocate elsewhere 2021-02-09 13:16:31 +09:00
embedding_lookup.cpp Update c++ copyrights 2021-02-01 21:31:45 +09:00
fusedBatchNorm.cpp Fix NCHW case for fused batch norm 2021-02-16 11:02:27 +09:00
layer_norm.cpp Update copyrights remove attic and relocate elsewhere 2021-02-09 13:16:31 +09:00
logSoftmax.cpp Update c++ copyrights 2021-02-01 21:31:45 +09:00
lrn.cpp Update c++ copyrights 2021-02-01 21:31:45 +09:00
multi_head_dot_product_attention.cpp Update copyrights remove attic and relocate elsewhere 2021-02-09 13:16:31 +09:00
relu_layer.cpp Update c++ copyrights 2021-02-01 21:31:45 +09:00
softmax.cpp Update c++ copyrights 2021-02-01 21:31:45 +09:00
xw_plus_b.cpp Update c++ copyrights 2021-02-01 21:31:45 +09:00