Python get_inter_op_parallelism_threads Examples

Example #1

File: config_test.py Project: zuozi2810/tensorflow

    def testInterOpParallelismThreads(self):
        config.set_inter_op_parallelism_threads(10)
        self.assertEqual(config.get_inter_op_parallelism_threads(),
                         context.context().inter_op_parallelism_threads)

        constant_op.constant(1)
        with self.assertRaises(RuntimeError):
            config.set_inter_op_parallelism_threads(1)

Example #2

File: config_test.py Project: kylin9872/tensorflow

  def testInterOpParallelismThreads(self):
    config.set_inter_op_parallelism_threads(10)
    self.assertEqual(
        config.get_inter_op_parallelism_threads(),
        context.context().inter_op_parallelism_threads)

    constant_op.constant(1)
    with self.assertRaises(RuntimeError):
      config.set_inter_op_parallelism_threads(1)

Example #3

File: config_test.py Project: waterdrops/tensorflow

    def testInterOpParallelismThreads(self):
        config.set_inter_op_parallelism_threads(10)
        self.assertEqual(config.get_inter_op_parallelism_threads(),
                         context.context().inter_op_parallelism_threads)

        context.ensure_initialized()

        with self.assertRaises(RuntimeError):
            config.set_inter_op_parallelism_threads(1)

        config.set_inter_op_parallelism_threads(10)