Internal change

PiperOrigin-RevId: 295821819
Change-Id: I7307e94062e3020ec26896634cfc23041773ff8e
This commit is contained in:
A. Unique TensorFlower 2020-02-18 14:23:36 -08:00 committed by TensorFlower Gardener
parent 79fdda5ead
commit b391cb55c2
1 changed files with 2 additions and 0 deletions

View File

@ -124,6 +124,8 @@ class Conv(Layer):
activity_regularizer=regularizers.get(activity_regularizer),
**kwargs)
self.rank = rank
if filters is not None and not isinstance(filters, int):
filters = int(filters)
self.filters = filters
self.kernel_size = conv_utils.normalize_tuple(
kernel_size, rank, 'kernel_size')