Commit 7f1c0af6 authored by opensourcemattress's avatar opensourcemattress Committed by gunan
Browse files

Float16 (half or Eigen::half) for conv3d ops (#12832)

* fp16 on conv3d, conv3d_transpose, tf.layers.batch_norm and tf.slim.batch_norm (batchnorms without fused_batch_norm)

* fp16 for conv3d_transposed

* remove batch_norm fp16 and rewrite conv3d kernels registrations

* remove additional logging prints

* tests for conv3d gradients

* fp64 test for cpu

* input normalization and requests

* fix initialization comment and remove unnecessary variable

* lowered tolerance for fp16 test

* remove unnecessary part of the comment
parent 69ba0d89
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment