Skip to content

Commit

Permalink
Skip channels_first with TF
Browse files Browse the repository at this point in the history
  • Loading branch information
fchollet committed Jul 17, 2023
1 parent 401ec43 commit 2a43b97
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions keras_core/layers/convolutional/conv_transpose_test.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,9 @@
import numpy as np
import pytest
import tensorflow as tf
from absl.testing import parameterized

from keras_core import backend
from keras_core import layers
from keras_core import testing

Expand Down Expand Up @@ -131,6 +133,12 @@ def test_conv2d_transpose_basic(
input_shape,
output_shape,
):
if (
data_format == "channels_first"
and backend.backend() == "tensorflow"
):
pytest.skip("channels_first unsupported on CPU with TF")

self.run_layer_test(
layers.Conv2DTranspose,
init_kwargs={
Expand Down

0 comments on commit 2a43b97

Please sign in to comment.