def forward(self, x): x = self.conv1(x) x = nn.tf_gelu(x) x = nn.tf_depth_to_space(x, 2) return x
def forward(self, x): x = self.conv1(x) x = tf.nn.leaky_relu(x, 0.1) x = nn.tf_depth_to_space(x, 2) return x