Internal change
PiperOrigin-RevId: 295821819 Change-Id: I7307e94062e3020ec26896634cfc23041773ff8e
This commit is contained in:
parent
79fdda5ead
commit
b391cb55c2
|
@ -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')
|
||||
|
|
Loading…
Reference in New Issue