search

amazon-archives / amazon-dsstne

Deep Scalable Sparse Tensor Network Engine (DSSTNE) is an Amazon developed library for building Deep Learning (DL) machine learning (ML) models
Apache License 2.0
4.41k stars 731 forks source link

Python-C++ extension functions that support dsstne API via Python 'dsstne' module #213

Closed chezruss closed 5 years ago

chezruss commented 5 years ago

Description of changes:

Creation of the amazon-dsstne/python directory that contains the source and test code for the Python 'dsstne' module.

Minor changes to the following files in the amazon-dsstne/src/amazon/dsstne/engine directory in order to support the 'dsstne' module: GpuTypes.cpp, NNLayer.cpp, NNLayer.h, NNNetwork.cpp, NNNetwork.h, NNTypes.cpp, NNTypes.h, NNWeight.cpp, and NNWeight.h

Add '#ifndef ... #define ... #endif' directives to amazon-dsstne/src/amazon/dsstne/utils/cdl.h in order to prevent inadvertent inclusion of more than one copy of cdl.h

Add comments to amazon-dsstne/samples/cifar-10/dparse.cpp

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.