From 63501a0b5666a74f8e303c75eca2f4eb2899244b Mon Sep 17 00:00:00 2001 From: Kaan <46622558+Frightera@users.noreply.github.com> Date: Sun, 8 Aug 2021 19:28:04 +0300 Subject: [PATCH] Added input_shape param when testing layers --- tensorflow_probability/python/layers/conv_variational_test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow_probability/python/layers/conv_variational_test.py b/tensorflow_probability/python/layers/conv_variational_test.py index e1aaa9afd7..90d5175f89 100644 --- a/tensorflow_probability/python/layers/conv_variational_test.py +++ b/tensorflow_probability/python/layers/conv_variational_test.py @@ -609,7 +609,8 @@ def _testLayerInSequential(self, layer_class): # pylint: disable=invalid-name outputs = self.maybe_transpose_tensor(outputs) net = tf.keras.Sequential([ - layer_class(filters=2, kernel_size=3, data_format=self.data_format), + layer_class(filters=2, kernel_size=3, data_format=self.data_format, + input_shape = inputs.shape.as_list()[1:]), layer_class(filters=2, kernel_size=1, data_format=self.data_format)]) net.compile(loss='mse', optimizer='adam')