Searched refs:_assertOpAndComposite (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | mnist_ops_test.py | 52 self._assertOpAndComposite([input_, filter_, bias],
72 self._assertOpAndComposite([input_, filter_, bias],
93 self._assertOpAndComposite([input_, filter_, bias],
103 self._assertOpAndComposite([input_, filter_, bias],
118 self._assertOpAndComposite([input_],
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/ |
| D | pad_ops_test.py | 56 self._assertOpAndComposite([input_], tf.raw_ops.MirrorPad,
59 self._assertOpAndComposite([input_],
84 self._assertOpAndComposite([input_], tf.raw_ops.MirrorPadGrad,
88 self._assertOpAndComposite([input_],
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | test_utils.py | 26 def _assertOpAndComposite(self, vars_, compute_op, compute_composite, kwargs, member in OpsDefsTest
|