'bias': bias,
        'stride_w': 2,
        'stride_h': 2,
        'dilation_w': 1,
        'dilation_h': 1,
        'padding': 'SAME',
        'act': 'RELU'
    }

    self._assertOpAndComposite([input_, filter_, bias],
                               tf.function(gen_mnist_ops.new_conv2d),
                               ops_defs._composite_conv_add_relu, kwargs)

  def test_new_conv2d_relu6(self):

        'mode': mode,
    }
    # Make sure the composition python function is correct
    self._assertOpAndComposite([input_], tf.raw_ops.MirrorPad,
                               ops_defs._composite_mirror_pad, kwargs_, kwargs)
    # Make sure the translation and decomposition is correct
    self._assertOpAndComposite([input_],
                               tf.function(gen_pad_ops.new_mirror_pad),

from tensorflow.python.eager import backprop
from tensorflow.python.framework import test_util
from tensorflow.python.platform import test


class OpsDefsTest(test.TestCase):
  """Test utils."""

  def _assertOpAndComposite(self, vars_, compute_op, compute_composite, kwargs,
                            op_kwargs=None):
    if op_kwargs is None:
      op_kwargs = kwargs
    if test_util.IsMklEnabled():