Ejemplos de set_global_constants en Python

Ejemplo n.º 1

Archivo: fisher_factors_test.py Proyecto: QiangCai/tensorflow

 def testTrue(self):
   ff.set_global_constants(colocate_cov_ops_with_inputs=True)
   with tf_ops.Graph().as_default():
     a = constant_op.constant([2.0], name='a')
     with ff.maybe_colocate_with(a):
       b = constant_op.constant(3.0, name='b')
     self.assertEqual([b'loc:@a'], a.op.colocation_groups())
     self.assertEqual([b'loc:@a'], b.op.colocation_groups())

Ejemplo n.º 2

Archivo: fisher_factors_test.py Proyecto: ViralLeadership/Repositorios

 def testTrue(self):
     ff.set_global_constants(colocate_cov_ops_with_inputs=True)
     with tf_ops.Graph().as_default():
         a = constant_op.constant([2.0], name='a')
         with ff.maybe_colocate_with(a):
             b = constant_op.constant(3.0, name='b')
         self.assertEqual([b'loc:@a'], a.op.colocation_groups())
         self.assertEqual([b'loc:@a'], b.op.colocation_groups())

Ejemplo n.º 3

Archivo: optimizer_test.py Proyecto: yytian9/tensorflow

from tensorflow.contrib.kfac.python.ops import fisher_factors as ff
from tensorflow.contrib.kfac.python.ops import layer_collection as lc
from tensorflow.contrib.kfac.python.ops import optimizer
from tensorflow.python.framework import ops
from tensorflow.python.ops import array_ops
from tensorflow.python.ops import init_ops
from tensorflow.python.ops import math_ops
from tensorflow.python.ops import nn
from tensorflow.python.ops import variable_scope
from tensorflow.python.ops import variables as tf_variables
from tensorflow.python.platform import test

# We need to set these constants since the numerical values used in the tests
# were chosen when these used to be the defaults.
ff.set_global_constants(init_covariances_at_zero=False,
                        zero_debias=False,
                        init_inverses_at_zero=False)


def dummy_layer_collection():
    lcoll = lc.LayerCollection()
    dummy = array_ops.constant([1., 2.])
    lcoll.register_categorical_predictive_distribution(logits=dummy)
    return lcoll


class OptimizerTest(test.TestCase):
    def testOptimizerInitInvalidMomentumRegistration(self):
        with self.assertRaises(ValueError):
            optimizer.KfacOptimizer(0.1,
                                    0.2,

Ejemplo n.º 4

Archivo: optimizer_test.py Proyecto: BhaskarNallani/tensorflow

from tensorflow.contrib.kfac.python.ops import layer_collection as lc
from tensorflow.contrib.kfac.python.ops import optimizer
from tensorflow.python.framework import ops
from tensorflow.python.ops import array_ops
from tensorflow.python.ops import init_ops
from tensorflow.python.ops import math_ops
from tensorflow.python.ops import nn
from tensorflow.python.ops import variable_scope
from tensorflow.python.ops import variables as tf_variables
from tensorflow.python.platform import test


# We need to set these constants since the numerical values used in the tests
# were chosen when these used to be the defaults.
ff.set_global_constants(init_covariances_at_zero=False,
                        zero_debias=False,
                        init_inverses_at_zero=False)


def dummy_layer_collection():
  lcoll = lc.LayerCollection()
  dummy = array_ops.constant([1., 2.])
  lcoll.register_categorical_predictive_distribution(logits=dummy)
  return lcoll


class OptimizerTest(test.TestCase):

  def testOptimizerInitInvalidMomentumRegistration(self):
    with self.assertRaises(ValueError):
      optimizer.KfacOptimizer(

Ejemplo n.º 5

Archivo: fisher_factors_test.py Proyecto: QiangCai/tensorflow

 def tearDown(self):
   ff.set_global_constants(
       colocate_cov_ops_with_inputs=self._colocate_cov_ops_with_inputs)

Ejemplo n.º 6

Archivo: fisher_factors_test.py Proyecto: ViralLeadership/Repositorios

 def tearDown(self):
     ff.set_global_constants(
         colocate_cov_ops_with_inputs=self._colocate_cov_ops_with_inputs)