def test_converted_call_already_converted(self): def f(x): return x == 0 with self.test_session() as sess: x = api.converted_call(f, False, False, {}, constant_op.constant(0)) self.assertTrue(sess.run(x)) converted_f = api.to_graph(f) x = api.converted_call(converted_f, False, False, {}, constant_op.constant(0)) self.assertTrue(sess.run(x))
def test_converted_call_already_converted(self): def f(x): return x == 0 with self.test_session() as sess: x = api.converted_call(f, False, False, False, {}, constant_op.constant(0)) self.assertTrue(sess.run(x)) converted_f = api.to_graph(f) x = api.converted_call(converted_f, False, False, False, {}, constant_op.constant(0)) self.assertTrue(sess.run(x))
def test_converted_call_function(self): def test_fn(x): if x < 0: return -x return x with self.test_session() as sess: x = api.converted_call(test_fn, False, False, False, {}, constant_op.constant(-1)) self.assertEqual(1, sess.run(x))
def test_converted_call_function(self): def test_fn(x): if x < 0: return -x return x with self.test_session() as sess: x = api.converted_call( test_fn, False, False, {}, constant_op.constant(-1)) self.assertEqual(1, sess.run(x))
def test_converted_call_callable_object(self): class TestClass(object): def __init__(self, x): self.x = x def __call__(self): if self.x < 0: return -self.x return self.x with self.test_session() as sess: tc = TestClass(constant_op.constant(-1)) x = api.converted_call(tc, False, False, False, {}) self.assertEqual(1, sess.run(x))
def test_converted_call_method_by_class(self): class TestClass(object): def __init__(self, x): self.x = x def test_method(self): if self.x < 0: return -self.x return self.x with self.test_session() as sess: tc = TestClass(constant_op.constant(-1)) x = api.converted_call(TestClass.test_method, False, False, {}, tc) self.assertEqual(1, sess.run(x))
def test_converted_call_constructor(self): class TestClass(object): def __init__(self, x): self.x = x def test_method(self): if self.x < 0: return -self.x return self.x with self.test_session() as sess: tc = api.converted_call(TestClass, False, False, False, {}, constant_op.constant(-1)) # tc is now a converted object. x = tc.test_method() self.assertEqual(1, sess.run(x))
def test_converted_call_callable_object(self): class TestClass(object): def __init__(self, x): self.x = x def __call__(self): if self.x < 0: return -self.x return self.x with self.test_session() as sess: tc = TestClass(constant_op.constant(-1)) x = api.converted_call(tc, False, False, {}) self.assertEqual(1, sess.run(x))
def test_converted_call_method_by_class(self): class TestClass(object): def __init__(self, x): self.x = x def test_method(self): if self.x < 0: return -self.x return self.x with self.test_session() as sess: tc = TestClass(constant_op.constant(-1)) x = api.converted_call(TestClass.test_method, False, False, {}, tc) self.assertEqual(1, sess.run(x))
def test_converted_call_constructor(self): class TestClass(object): def __init__(self, x): self.x = x def test_method(self): if self.x < 0: return -self.x return self.x with self.test_session() as sess: tc = api.converted_call( TestClass, False, False, {}, constant_op.constant(-1)) # tc is now a converted object. x = tc.test_method() self.assertEqual(1, sess.run(x))
def test_method(self, x, s, a): while tf.reduce_sum(x) > s: x //= api.converted_call(self.called_member, False, False, {}, self, a) return x
def test_converted_call_builtin(self): x = api.converted_call(range, False, False, False, {}, 3) self.assertEqual((0, 1, 2), tuple(x))
def test_converted_call_builtin(self): x = api.converted_call(range, False, False, {}, 3) self.assertEqual((0, 1, 2), tuple(x))
def test_method(self, x, s, a): while tf.reduce_sum(x) > s: x //= api.converted_call(self.called_member, False, False, {}, self, a) return x