D | tf_math_ops_test.py | 51 def test_1d(op_name, fn, vectorize=False, lb=-1.0, ub=1.0, rtol_enum=Rtol.BASE): argument 52 compiled = jitrt.compile(mlir_func_1d(op_name), 'test', vectorize=vectorize) 95 def test_op(self, op_name, fn, vectorize, rtol_enum): argument 96 test_1d(op_name, fn, vectorize=vectorize, rtol_enum=rtol_enum)
|