Open waylonflinn opened 8 years ago
The GLSL shader for Tensor.reshape is very complex. It also currently accounts for ~15% of the execution time of the NN demo.
Tensor.reshape
Some progress on this. Created an optimized (but un-benchmarked) version for the no padding (multiple of four on input and output) case.
The GLSL shader for
Tensor.reshape
is very complex. It also currently accounts for ~15% of the execution time of the NN demo.