Esempi in Python per get_variable_ddi

Esempio n. 1

File: glow_ops_test.py Progetto: y12uc231/BERT-1

 def test_get_variable_ddi(self):
     with tf.Graph().as_default():
         x_t = tf.random_normal((5, 5))
         ddi = glow_ops.get_variable_ddi("x", (5, 5), x_t, init=True)
         with tf.Session() as session:
             diff = ddi - x_t
             self.assertTrue(np.allclose(session.run(diff), 0.0))

Esempio n. 2

File: glow_ops_test.py Progetto: qixiuai/tensor2tensor

 def test_get_variable_ddi(self):
   with tf.Graph().as_default():
     x_t = tf.random_normal((5, 5))
     ddi = glow_ops.get_variable_ddi(
         "x", (5, 5), initial_value=x_t, init=True)
     with tf.Session() as session:
       diff = ddi - x_t
       self.assertTrue(np.allclose(session.run(diff), 0.0))