@@ -469,8 +469,9 @@ layer_separable_conv_2d <- function(object, filters, kernel_size, strides = c(1L
469
469
depth_multiplier = 1L , activation = NULL , use_bias = TRUE ,
470
470
depthwise_initializer = " glorot_uniform" , pointwise_initializer = " glorot_uniform" , bias_initializer = " zeros" ,
471
471
depthwise_regularizer = NULL , pointwise_regularizer = NULL , bias_regularizer = NULL , activity_regularizer = NULL ,
472
- depthwise_constraint = NULL , pointwise_constraint = NULL , bias_constraint = NULL ,
473
- batch_size = NULL , name = NULL , trainable = NULL , weights = NULL ) {
472
+ depthwise_constraint = NULL , pointwise_constraint = NULL , bias_constraint = NULL , input_shape = NULL ,
473
+ batch_input_shape = NULL , batch_size = NULL , dtype = NULL ,
474
+ name = NULL , trainable = NULL , weights = NULL ) {
474
475
475
476
create_layer(keras $ layers $ SeparableConv2D , object , list (
476
477
filters = as.integer(filters ),
@@ -491,7 +492,10 @@ layer_separable_conv_2d <- function(object, filters, kernel_size, strides = c(1L
491
492
depthwise_constraint = depthwise_constraint ,
492
493
pointwise_constraint = pointwise_constraint ,
493
494
bias_constraint = bias_constraint ,
495
+ input_shape = normalize_shape(input_shape ),
496
+ batch_input_shape = normalize_shape(batch_input_shape ),
494
497
batch_size = as_nullable_integer(batch_size ),
498
+ dtype = dtype ,
495
499
name = name ,
496
500
trainable = trainable ,
497
501
weights = weights
0 commit comments