diff options
Diffstat (limited to 'tensorflow/python/layers/convolutional.py')
| -rw-r--r-- | tensorflow/python/layers/convolutional.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/layers/convolutional.py b/tensorflow/python/layers/convolutional.py index 0c7ce02835..8c327d7e27 100644 --- a/tensorflow/python/layers/convolutional.py +++ b/tensorflow/python/layers/convolutional.py @@ -813,6 +813,7 @@ def conv3d(inputs, bias_constraint=bias_constraint, trainable=trainable, name=name, + dtype=inputs.dtype.base_dtype, _reuse=reuse, _scope=name) return layer.apply(inputs) @@ -1746,6 +1747,7 @@ def conv3d_transpose(inputs, bias_constraint=bias_constraint, trainable=trainable, name=name, + dtype=inputs.dtype.base_dtype, _reuse=reuse, _scope=name) return layer.apply(inputs) |