fix tests for channels_first (keras-team#18723)

* fix tests for channels_first * bug fix
Faisal-Alsrheed · Nov 4, 2023 · aa02010 · aa02010
1 parent c1ec2aa
commit aa02010
Show file tree

Hide file tree

Showing 2 changed files with 504 additions and 395 deletions.
diff --git a/keras/layers/convolutional/conv_transpose_test.py b/keras/layers/convolutional/conv_transpose_test.py
@@ -122,7 +122,7 @@ def np_conv2d_transpose(
         strides,
         padding,
         output_padding,
-        data_format,
+        "channels_last",
         dilation_rate,
     )
     jax_padding = compute_conv_transpose_padding_args_for_jax(