LICENSE
README.md
pyproject.toml
setup.py
pydtnn/__init__.py
pydtnn/layer_base.py
pydtnn/model.py
pydtnn/parser.py
pydtnn/pydtnn_benchmark.py
pydtnn.egg-info/PKG-INFO
pydtnn.egg-info/SOURCES.txt
pydtnn.egg-info/dependency_links.txt
pydtnn.egg-info/entry_points.txt
pydtnn.egg-info/requires.txt
pydtnn.egg-info/top_level.txt
pydtnn/activations/activation.py
pydtnn/activations/arctanh.py
pydtnn/activations/leaky_relu.py
pydtnn/activations/log.py
pydtnn/activations/relu.py
pydtnn/activations/relu6.py
pydtnn/activations/sigmoid.py
pydtnn/activations/softmax.py
pydtnn/activations/tanh.py
pydtnn/backends/__init__.py
pydtnn/backends/cupy/activations/activation.py
pydtnn/backends/cupy/activations/leaky_relu.py
pydtnn/backends/cupy/activations/relu.py
pydtnn/backends/cupy/activations/relu6.py
pydtnn/backends/cupy/layers/adaptive_average_pool_2d.py
pydtnn/backends/cupy/layers/average_pool_2d.py
pydtnn/backends/cupy/layers/batch_normalization.py
pydtnn/backends/cupy/layers/conv_2d.py
pydtnn/backends/cupy/layers/conv_2d_depthwise.py
pydtnn/backends/cupy/layers/fc.py
pydtnn/backends/cupy/layers/layer.py
pydtnn/backends/cupy/layers/max_pool_2d.py
pydtnn/backends/cupy/metrics/binary_confusion_matrix.py
pydtnn/backends/cupy/metrics/f1_score.py
pydtnn/backends/cupy/metrics/metric.py
pydtnn/backends/cupy/metrics/multiclass_confusion_matrix.py
pydtnn/backends/cupy/metrics/precision.py
pydtnn/backends/cupy/metrics/recall.py
pydtnn/backends/cython/activations/activation.py
pydtnn/backends/cython/activations/leaky_relu.py
pydtnn/backends/cython/activations/log.py
pydtnn/backends/cython/activations/relu.py
pydtnn/backends/cython/activations/relu6.py
pydtnn/backends/cython/activations/sigmoid.py
pydtnn/backends/cython/layers/adaptive_average_pool_2d.py
pydtnn/backends/cython/layers/average_pool_2d.py
pydtnn/backends/cython/layers/batch_normalization.py
pydtnn/backends/cython/layers/conv_2d.py
pydtnn/backends/cython/layers/conv_2d_depthwise.py
pydtnn/backends/cython/layers/max_pool_2d.py
pydtnn/backends/cython/utils/adaptive_avg_pooling_nchw_cython.c
pydtnn/backends/cython/utils/adaptive_avg_pooling_nchw_cython.pyi
pydtnn/backends/cython/utils/adaptive_avg_pooling_nhwc_cython.c
pydtnn/backends/cython/utils/adaptive_avg_pooling_nhwc_cython.pyi
pydtnn/backends/cython/utils/argmax_cython.c
pydtnn/backends/cython/utils/argmax_cython.pyi
pydtnn/backends/cython/utils/average_pool_2d_nchw_cython.c
pydtnn/backends/cython/utils/average_pool_2d_nchw_cython.pyi
pydtnn/backends/cython/utils/average_pool_2d_nhwc_cython.c
pydtnn/backends/cython/utils/average_pool_2d_nhwc_cython.pyi
pydtnn/backends/cython/utils/bn_training_cython.c
pydtnn/backends/cython/utils/bn_training_cython.pyi
pydtnn/backends/cython/utils/depthwise_conv_nchw_cython.c
pydtnn/backends/cython/utils/depthwise_conv_nchw_cython.pyi
pydtnn/backends/cython/utils/depthwise_conv_nhwc_cython.c
pydtnn/backends/cython/utils/depthwise_conv_nhwc_cython.pyi
pydtnn/backends/cython/utils/div_arrays_set_if_zero.c
pydtnn/backends/cython/utils/div_arrays_set_if_zero.pyi
pydtnn/backends/cython/utils/im2col_1ch_nchw_cython.c
pydtnn/backends/cython/utils/im2col_1ch_nchw_cython.pyi
pydtnn/backends/cython/utils/im2col_nchw_cython.c
pydtnn/backends/cython/utils/im2col_nchw_cython.pyi
pydtnn/backends/cython/utils/im2row_1ch_nhwc_cython.c
pydtnn/backends/cython/utils/im2row_1ch_nhwc_cython.pyi
pydtnn/backends/cython/utils/im2row_nhwc_cython.c
pydtnn/backends/cython/utils/im2row_nhwc_cython.pyi
pydtnn/backends/cython/utils/log_activation_cython.c
pydtnn/backends/cython/utils/log_activation_cython.pyi
pydtnn/backends/cython/utils/max_pool_2d_nchw_cython.c
pydtnn/backends/cython/utils/max_pool_2d_nchw_cython.pyi
pydtnn/backends/cython/utils/max_pool_2d_nhwc_cython.c
pydtnn/backends/cython/utils/max_pool_2d_nhwc_cython.pyi
pydtnn/backends/cython/utils/oktopk_utils_cython.c
pydtnn/backends/cython/utils/oktopk_utils_cython.pyi
pydtnn/backends/cython/utils/pointwise_conv_cython.c
pydtnn/backends/cython/utils/pointwise_conv_cython.pyi
pydtnn/backends/cython/utils/relu_cython.c
pydtnn/backends/cython/utils/relu_cython.pyi
pydtnn/backends/cython/utils/sigmoid_cython.c
pydtnn/backends/cython/utils/sigmoid_cython.pyi
pydtnn/backends/direct/layers/conv_2d.py
pydtnn/backends/fuse/layers/__init__.py
pydtnn/backends/fuse/layers/batch_normalization_relu.py
pydtnn/backends/fuse/layers/conv_2d_batch_normalization.py
pydtnn/backends/fuse/layers/conv_2d_batch_normalization_relu.py
pydtnn/backends/fuse/layers/conv_2d_relu.py
pydtnn/backends/fuse/utils/bn_inference_cython.c
pydtnn/backends/fuse/utils/bn_inference_cython.pyi
pydtnn/backends/gemm/layers/conv_2d.py
pydtnn/backends/numpy/activations/activation.py
pydtnn/backends/numpy/activations/arctanh.py
pydtnn/backends/numpy/activations/leaky_relu.py
pydtnn/backends/numpy/activations/log.py
pydtnn/backends/numpy/activations/relu.py
pydtnn/backends/numpy/activations/relu6.py
pydtnn/backends/numpy/activations/sigmoid.py
pydtnn/backends/numpy/activations/softmax.py
pydtnn/backends/numpy/activations/tanh.py
pydtnn/backends/numpy/layers/adaptive_average_pool_2d.py
pydtnn/backends/numpy/layers/addition_block.py
pydtnn/backends/numpy/layers/average_pool_2d.py
pydtnn/backends/numpy/layers/batch_normalization.py
pydtnn/backends/numpy/layers/concatenation_block.py
pydtnn/backends/numpy/layers/conv_2d.py
pydtnn/backends/numpy/layers/conv_2d_depthwise.py
pydtnn/backends/numpy/layers/conv_2d_pointwise.py
pydtnn/backends/numpy/layers/decoder.py
pydtnn/backends/numpy/layers/dropout.py
pydtnn/backends/numpy/layers/encoder.py
pydtnn/backends/numpy/layers/encoder_decoder.py
pydtnn/backends/numpy/layers/fc.py
pydtnn/backends/numpy/layers/feed_forward.py
pydtnn/backends/numpy/layers/flatten.py
pydtnn/backends/numpy/layers/input.py
pydtnn/backends/numpy/layers/layer.py
pydtnn/backends/numpy/layers/layer_normalization.py
pydtnn/backends/numpy/layers/max_pool_2d.py
pydtnn/backends/numpy/layers/multi_head_attention.py
pydtnn/backends/numpy/layers/multiplication.py
pydtnn/backends/numpy/layers/scalar.py
pydtnn/backends/numpy/layers/abstract/block_layer.py
pydtnn/backends/numpy/layers/abstract/conv_2d.py
pydtnn/backends/numpy/layers/abstract/conv_2d_standard.py
pydtnn/backends/numpy/layers/abstract/pool_2d_layer.py
pydtnn/backends/numpy/losses/binary_cross_entropy.py
pydtnn/backends/numpy/losses/categorical_cross_entropy.py
pydtnn/backends/numpy/losses/kl_divergence.py
pydtnn/backends/numpy/losses/loss.py
pydtnn/backends/numpy/metrics/binary_confusion_matrix.py
pydtnn/backends/numpy/metrics/categorical_accuracy.py
pydtnn/backends/numpy/metrics/categorical_hinge.py
pydtnn/backends/numpy/metrics/categorical_mae.py
pydtnn/backends/numpy/metrics/categorical_mse.py
pydtnn/backends/numpy/metrics/f1_score.py
pydtnn/backends/numpy/metrics/kl_divergence_metric.py
pydtnn/backends/numpy/metrics/metric.py
pydtnn/backends/numpy/metrics/multiclass_confusion_matrix.py
pydtnn/backends/numpy/metrics/precision.py
pydtnn/backends/numpy/metrics/recall.py
pydtnn/backends/numpy/metrics/regression_mae.py
pydtnn/backends/numpy/metrics/regression_mse.py
pydtnn/backends/numpy/optimizers/adam.py
pydtnn/backends/numpy/optimizers/nadam.py
pydtnn/backends/numpy/optimizers/oktopk.py
pydtnn/backends/numpy/optimizers/optimizer.py
pydtnn/backends/numpy/optimizers/rmsprop.py
pydtnn/backends/numpy/optimizers/sgd.py
pydtnn/backends/pycuda/activations/activation.py
pydtnn/backends/pycuda/activations/arctanh.py
pydtnn/backends/pycuda/activations/leaky_relu.py
pydtnn/backends/pycuda/activations/log.py
pydtnn/backends/pycuda/activations/relu.py
pydtnn/backends/pycuda/activations/relu6.py
pydtnn/backends/pycuda/activations/sigmoid.py
pydtnn/backends/pycuda/activations/softmax.py
pydtnn/backends/pycuda/activations/tanh.py
pydtnn/backends/pycuda/layers/adaptive_average_pool_2d.py
pydtnn/backends/pycuda/layers/addition_block.py
pydtnn/backends/pycuda/layers/average_pool_2d.py
pydtnn/backends/pycuda/layers/batch_normalization.py
pydtnn/backends/pycuda/layers/concatenation_block.py
pydtnn/backends/pycuda/layers/conv_2d.py
pydtnn/backends/pycuda/layers/conv_2d_depthwise.py
pydtnn/backends/pycuda/layers/conv_2d_pointwise.py
pydtnn/backends/pycuda/layers/decoder.py
pydtnn/backends/pycuda/layers/dropout.py
pydtnn/backends/pycuda/layers/encoder.py
pydtnn/backends/pycuda/layers/encoder_decoder.py
pydtnn/backends/pycuda/layers/fc.py
pydtnn/backends/pycuda/layers/feed_forward.py
pydtnn/backends/pycuda/layers/flatten.py
pydtnn/backends/pycuda/layers/input.py
pydtnn/backends/pycuda/layers/layer.py
pydtnn/backends/pycuda/layers/layer_normalization.py
pydtnn/backends/pycuda/layers/max_pool_2d.py
pydtnn/backends/pycuda/layers/multi_head_attention.py
pydtnn/backends/pycuda/layers/abstract/block_layer.py
pydtnn/backends/pycuda/layers/abstract/conv_2d.py
pydtnn/backends/pycuda/layers/abstract/pool_2d_layer.py
pydtnn/backends/pycuda/layers/abstract/standard_pycuda.py
pydtnn/backends/pycuda/losses/binary_cross_entropy.py
pydtnn/backends/pycuda/losses/categorical_cross_entropy.py
pydtnn/backends/pycuda/losses/kl_divergence.py
pydtnn/backends/pycuda/losses/loss.py
pydtnn/backends/pycuda/metrics/binary_confusion_matrix.py
pydtnn/backends/pycuda/metrics/categorical_accuracy.py
pydtnn/backends/pycuda/metrics/categorical_hinge.py
pydtnn/backends/pycuda/metrics/categorical_mae.py
pydtnn/backends/pycuda/metrics/categorical_mse.py
pydtnn/backends/pycuda/metrics/f1_score.py
pydtnn/backends/pycuda/metrics/kl_divergence_metric.py
pydtnn/backends/pycuda/metrics/metric.py
pydtnn/backends/pycuda/metrics/multiclass_confusion_matrix.py
pydtnn/backends/pycuda/metrics/precision.py
pydtnn/backends/pycuda/metrics/recall.py
pydtnn/backends/pycuda/metrics/regression_mae.py
pydtnn/backends/pycuda/metrics/regression_mse.py
pydtnn/backends/pycuda/optimizers/adam.py
pydtnn/backends/pycuda/optimizers/nadam.py
pydtnn/backends/pycuda/optimizers/optimizer.py
pydtnn/backends/pycuda/optimizers/rmsprop.py
pydtnn/backends/pycuda/optimizers/sgd.py
pydtnn/backends/pycuda/utils/__init__.py
pydtnn/backends/pycuda/utils/memory_allocation.py
pydtnn/backends/pycuda/utils/tensor_array.py
pydtnn/backends/winograd/layers/conv_2d.py
pydtnn/converters/onnx2pydtnn/constants.py
pydtnn/converters/onnx2pydtnn/model_converter.py
pydtnn/converters/onnx2pydtnn/operations/implemented_operations.py
pydtnn/converters/pydtnn2onnx/model_converter.py
pydtnn/converters/pytorch2pydtnn/common.py
pydtnn/converters/pytorch2pydtnn/model_convertor.py
pydtnn/converters/pytorch2pydtnn/test.py
pydtnn/converters/pytorch2pydtnn/test_conv.py
pydtnn/converters/pytorch2pydtnn/test_layers.py
pydtnn/converters/pytorch2pydtnn/layers/activation.py
pydtnn/converters/pytorch2pydtnn/layers/convolutional.py
pydtnn/converters/pytorch2pydtnn/layers/dropout.py
pydtnn/converters/pytorch2pydtnn/layers/functions.py
pydtnn/converters/pytorch2pydtnn/layers/linear.py
pydtnn/converters/pytorch2pydtnn/layers/normalization.py
pydtnn/converters/pytorch2pydtnn/layers/pooling.py
pydtnn/converters/pytorch2pydtnn/layers/utility.py
pydtnn/datasets/archive.py
pydtnn/datasets/chestxray.py
pydtnn/datasets/cifar10.py
pydtnn/datasets/cyclone.py
pydtnn/datasets/dataset.py
pydtnn/datasets/folder.py
pydtnn/datasets/imagenet.py
pydtnn/datasets/iwslt.py
pydtnn/datasets/mask_lang.py
pydtnn/datasets/memory.py
pydtnn/datasets/mnist.py
pydtnn/datasets/synthetic.py
pydtnn/datasets/tsunamis.py
pydtnn/layers/adaptive_average_pool_2d.py
pydtnn/layers/addition_block.py
pydtnn/layers/average_pool_2d.py
pydtnn/layers/batch_normalization.py
pydtnn/layers/concatenation_block.py
pydtnn/layers/conv_2d.py
pydtnn/layers/conv_2d_depthwise.py
pydtnn/layers/conv_2d_pointwise.py
pydtnn/layers/decoder.py
pydtnn/layers/dropout.py
pydtnn/layers/encoder.py
pydtnn/layers/encoder_decoder.py
pydtnn/layers/fc.py
pydtnn/layers/feed_forward.py
pydtnn/layers/flatten.py
pydtnn/layers/input.py
pydtnn/layers/layer.py
pydtnn/layers/layer_normalization.py
pydtnn/layers/max_pool_2d.py
pydtnn/layers/multi_head_attention.py
pydtnn/layers/multiplication.py
pydtnn/layers/scalar.py
pydtnn/layers/abstract/block_layer.py
pydtnn/layers/abstract/pool_2d_layer.py
pydtnn/libs/convDirect.py
pydtnn/libs/convGemm.py
pydtnn/libs/convWinograd.py
pydtnn/libs/cublas.py
pydtnn/libs/cuda.py
pydtnn/libs/cudadrv.py
pydtnn/libs/cudart.py
pydtnn/libs/cudnn.py
pydtnn/libs/nccl.py
pydtnn/libs/numpy.py
pydtnn/libs/utils.py
pydtnn/libs/mpi/MPI.py
pydtnn/libs/mpi/rc.py
pydtnn/losses/binary_cross_entropy.py
pydtnn/losses/categorical_cross_entropy.py
pydtnn/losses/kl_divergence.py
pydtnn/losses/loss.py
pydtnn/metrics/binary_confusion_matrix.py
pydtnn/metrics/categorical_accuracy.py
pydtnn/metrics/categorical_hinge.py
pydtnn/metrics/categorical_mae.py
pydtnn/metrics/categorical_mse.py
pydtnn/metrics/confusion_matrix.py
pydtnn/metrics/f1_score.py
pydtnn/metrics/kl_divergence_metric.py
pydtnn/metrics/metric.py
pydtnn/metrics/multiclass_confusion_matrix.py
pydtnn/metrics/precision.py
pydtnn/metrics/recall.py
pydtnn/metrics/regression_mae.py
pydtnn/metrics/regression_mse.py
pydtnn/models/alexnet.py
pydtnn/models/alexnet_cifar10.py
pydtnn/models/alexnet_imagenet.py
pydtnn/models/bert.py
pydtnn/models/densenet121.py
pydtnn/models/densenet121_cifar10.py
pydtnn/models/densenet121_imagenet.py
pydtnn/models/densenet161.py
pydtnn/models/densenet161_cifar10.py
pydtnn/models/densenet161_imagenet.py
pydtnn/models/densenet169.py
pydtnn/models/densenet169_cifar10.py
pydtnn/models/densenet169_imagenet.py
pydtnn/models/densenet201.py
pydtnn/models/densenet201_cifar10.py
pydtnn/models/densenet201_imagenet.py
pydtnn/models/densenet_cifar10.py
pydtnn/models/googlenet.py
pydtnn/models/googlenet_cifar10.py
pydtnn/models/googlenet_imagenet.py
pydtnn/models/inceptionv3.py
pydtnn/models/inceptionv3_cifar10.py
pydtnn/models/inceptionv3_imagenet.py
pydtnn/models/iwslt.py
pydtnn/models/mobilenetv1.py
pydtnn/models/mobilenetv1_cifar10.py
pydtnn/models/mobilenetv1_pytorch.py
pydtnn/models/mobilenetv1_tensorflow.py
pydtnn/models/mobilenetv2_pytorch.py
pydtnn/models/model.py
pydtnn/models/resnet.py
pydtnn/models/resnet101.py
pydtnn/models/resnet101_cifar10.py
pydtnn/models/resnet101_imagenet.py
pydtnn/models/resnet110.py
pydtnn/models/resnet110_cifar10.py
pydtnn/models/resnet1202.py
pydtnn/models/resnet1202_cifar10.py
pydtnn/models/resnet152.py
pydtnn/models/resnet152_cifar10.py
pydtnn/models/resnet152_imagenet.py
pydtnn/models/resnet18.py
pydtnn/models/resnet18_cifar10.py
pydtnn/models/resnet18_imagenet.py
pydtnn/models/resnet20.py
pydtnn/models/resnet20_cifar10.py
pydtnn/models/resnet32.py
pydtnn/models/resnet32_cifar10.py
pydtnn/models/resnet34.py
pydtnn/models/resnet34_cifar10.py
pydtnn/models/resnet34_imagenet.py
pydtnn/models/resnet44.py
pydtnn/models/resnet44_cifar10.py
pydtnn/models/resnet50.py
pydtnn/models/resnet50_cifar10.py
pydtnn/models/resnet50_imagenet.py
pydtnn/models/resnet50_pytorch.py
pydtnn/models/resnet50_tensorflow.py
pydtnn/models/resnet50v15_imagenet.py
pydtnn/models/resnet56.py
pydtnn/models/resnet56_cifar10.py
pydtnn/models/simplecnn.py
pydtnn/models/simplemlp.py
pydtnn/models/transformer_test.py
pydtnn/models/tsunamis_eflows_UMA.py
pydtnn/models/vgg1.py
pydtnn/models/vgg11.py
pydtnn/models/vgg11_cifar10.py
pydtnn/models/vgg11_imagenet.py
pydtnn/models/vgg11bn.py
pydtnn/models/vgg11bn_cifar10.py
pydtnn/models/vgg16.py
pydtnn/models/vgg16_cifar10.py
pydtnn/models/vgg16_imagenet.py
pydtnn/models/vgg16_pytorch.py
pydtnn/models/vgg16_tensorflow.py
pydtnn/models/vgg16bn.py
pydtnn/models/vgg16bn_cifar10.py
pydtnn/models/vgg19.py
pydtnn/models/vgg19_imagenet.py
pydtnn/models/vgg2.py
pydtnn/models/vgg3.py
pydtnn/models/vgg3do.py
pydtnn/models/vgg3do2.py
pydtnn/models/vgg3dobn.py
pydtnn/models/vgg_tropical_cyclone.py
pydtnn/optimizers/adam.py
pydtnn/optimizers/nadam.py
pydtnn/optimizers/oktopk.py
pydtnn/optimizers/optimizer.py
pydtnn/optimizers/rmsprop.py
pydtnn/optimizers/sgd.py
pydtnn/profilers/best_of_profiler.py
pydtnn/profilers/diff_profiles.py
pydtnn/profilers/extract_profile_from_output.py
pydtnn/profilers/profile_best_transpose_0231.py
pydtnn/profilers/profile_best_transpose_0312.py
pydtnn/profilers/profile_best_transpose_1023.py
pydtnn/profilers/profile_conv2d_conv_gemm.py
pydtnn/schedulers/early_stopping.py
pydtnn/schedulers/model_checkpoint.py
pydtnn/schedulers/reduce_lr_every_nepochs.py
pydtnn/schedulers/reduce_lr_on_plateau.py
pydtnn/schedulers/scheduler.py
pydtnn/schedulers/scheduler_with_loss_or_metric.py
pydtnn/schedulers/stop_at_loss.py
pydtnn/schedulers/warm_up.py
pydtnn/tests/batch_normalization_relu.py
pydtnn/tests/conv_2d_batch_normalization.py
pydtnn/tests/conv_2d_batch_normalization_relu.py
pydtnn/tests/conv_2d_conv_gemm.py
pydtnn/tests/conv_2d_conv_gemm_slow.py
pydtnn/tests/conv_2d_group.py
pydtnn/tests/conv_2d_relu.py
pydtnn/tests/conv_direct.py
pydtnn/tests/conv_gemm.py
pydtnn/tests/conv_winograd.py
pydtnn/tests/layer_pytorch.py
pydtnn/tests/model_conv_gemm.py
pydtnn/tests/model_dtype.py
pydtnn/tests/model_gpu.py
pydtnn/tests/model_tensor.py
pydtnn/tests/abstract/common.py
pydtnn/tests/abstract/conv_2d_common.py
pydtnn/tests/abstract/conv_common.py
pydtnn/tests/abstract/model_common.py
pydtnn/tests/groups/all.py
pydtnn/tests/groups/fused.py
pydtnn/tests/groups/layer.py
pydtnn/tests/groups/libs.py
pydtnn/tests/groups/model.py
pydtnn/tracers/events.py
pydtnn/tracers/extrae_tracer.py
pydtnn/tracers/simple_tracer.py
pydtnn/tracers/simple_tracer_gpu.py
pydtnn/tracers/simple_tracer_pmlib.py
pydtnn/tracers/tracer.py
pydtnn/utils/__init__.py
pydtnn/utils/constants.py
pydtnn/utils/debug.py
pydtnn/utils/gpu.py
pydtnn/utils/initializers.py
pydtnn/utils/matmul.py
pydtnn/utils/memory_cache.py
pydtnn/utils/memory_pool.py
pydtnn/utils/performance_counter.py
pydtnn/utils/performance_models.py
pydtnn/utils/pmlib.py
pydtnn/utils/profiler.py
pydtnn/utils/random.py
pydtnn/utils/sparse_cython.c
pydtnn/utils/tensor.py
pydtnn/utils/transpose_cython.c
pydtnn/utils/transpose_cython.pyi
pydtnn/utils/best_of/best_of.py
pydtnn/utils/best_of/best_of_variant.py
pydtnn/utils/best_of/best_transpose_0231.py
pydtnn/utils/best_of/best_transpose_0312.py
pydtnn/utils/best_of/best_transpose_1023.py
pydtnn/utils/sparse/sparse.py
pydtnn/utils/sparse/sparse_cython.c
pydtnn/utils/sparse/sparse_cython.pyi