Skip to content

SDNAFIO/tensorflow

 
 

About

This repository is a fork from the official tensorflow repository. Currently the following features have been added:

  • Support for grouped convolutions using cuDNN
    • accessible through the groups parameter in tf.layers.conv3d and tf.layers.conv2d
    • Please note that grouped convolutions are only supported on GPUs at the moment, on CPUs an error will be thrown.

Dependencies

The following environment has been used for testing:

  • Cuda 10.0
  • Cudnn 7.4.2
  • Python 3.6
  • Bazel 0.17.2

About

An Open Source Machine Learning Framework for Everyone

Resources

License

Code of conduct

Contributing

Security policy

Stars

Watchers

Forks

Packages

No packages published

Languages

  • C++ 49.4%
  • Python 40.4%
  • HTML 4.3%
  • Jupyter Notebook 2.4%
  • Go 1.3%
  • Java 0.8%
  • Other 1.4%