Python python_op Examples

Example #1

File: test_pure_ht.py Project: zhoutengteng/tdb

def build_graph_ht():
	a=tf.constant(2)
	b=tf.constant(3)
	c=tdb.python_op(myadd,inputs=[a,b],outputs=[tf.placeholder(tf.int32)])
	c2=tdb.python_op(mymult,inputs=[a,b],outputs=[tf.placeholder(tf.int32)])
	d=tdb.python_op(myneg,inputs=[c],outputs=[tf.placeholder(tf.int32)])
	return a,b,c,c2,d

Example #2

File: test_mixed.py Project: 21hub/tdb

	def test_1(self):
		"""
		ht->tf
		"""
		a=tf.constant(2)
		b=tf.constant(3)
		c=tdb.python_op(myadd,inputs=[a,b],outputs=[tf.placeholder(tf.int32)]) # a+b
		d=tf.neg(c)
		status,result=tdb.debug([d], feed_dict=None, breakpoints=None, break_immediately=False)	
		self.assertEqual(status, tdb.FINISHED)
		self.assertEqual(result[0],-5)

Example #3

File: test_mixed.py Project: 21hub/tdb

	def test_2(self):
		"""
		tf -> ht
		"""
		a=tf.constant(2)
		b=tf.constant(3)
		c=tf.add(a,b)
		d=tdb.python_op(myneg,inputs=[c],outputs=[tf.placeholder(tf.int32)])
		status,result=tdb.debug([d], feed_dict=None, breakpoints=None, break_immediately=False)
		self.assertEqual(status, tdb.FINISHED)
		self.assertEqual(result[0],-5)
		self.assertEqual(tdb.get_value(d),-5)
		self.assertEqual(tdb.get_value(d.name),-5)
		

Example #4

    def test_1(self):
        """
		ht->tf
		"""
        a = tf.constant(2)
        b = tf.constant(3)
        c = tdb.python_op(myadd,
                          inputs=[a, b],
                          outputs=[tf.placeholder(tf.int32)])  # a+b
        d = tf.neg(c)
        status, result = tdb.debug([d],
                                   feed_dict=None,
                                   breakpoints=None,
                                   break_immediately=False)
        self.assertEqual(status, tdb.FINISHED)
        self.assertEqual(result[0], -5)

Example #5

    def test_2(self):
        """
		tf -> ht
		"""
        a = tf.constant(2)
        b = tf.constant(3)
        c = tf.add(a, b)
        d = tdb.python_op(myneg,
                          inputs=[c],
                          outputs=[tf.placeholder(tf.int32)])
        status, result = tdb.debug([d],
                                   feed_dict=None,
                                   breakpoints=None,
                                   break_immediately=False)
        self.assertEqual(status, tdb.FINISHED)
        self.assertEqual(result[0], -5)
        self.assertEqual(tdb.get_value(d), -5)
        self.assertEqual(tdb.get_value(d.name), -5)