search

benoitsteiner / tensorflow-opencl

OpenCL support for TensorFlow
Apache License 2.0
472 stars 86 forks source link

Register batch normalization kernels for OpenCL #61

Closed ville-k closed 7 years ago

ville-k commented 7 years ago

Batch normalization for OpenCL/SYCL. Only registering float and double kernels - half was causing compiler warnings about not being implemented.