Python get_intra_op_parallelism_threadsの例

コード例 #1

ファイル: config_test.py プロジェクト: zuozi2810/tensorflow

    def testIntraOpParallelismThreads(self):
        config.set_intra_op_parallelism_threads(10)
        self.assertEqual(config.get_intra_op_parallelism_threads(),
                         context.context().intra_op_parallelism_threads)

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

コード例 #2

ファイル: config_test.py プロジェクト: kylin9872/tensorflow

  def testIntraOpParallelismThreads(self):
    config.set_intra_op_parallelism_threads(10)
    self.assertEqual(
        config.get_intra_op_parallelism_threads(),
        context.context().intra_op_parallelism_threads)

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

コード例 #3

ファイル: config_test.py プロジェクト: waterdrops/tensorflow

    def testIntraOpParallelismThreads(self):
        config.set_intra_op_parallelism_threads(10)
        self.assertEqual(config.get_intra_op_parallelism_threads(),
                         context.context().intra_op_parallelism_threads)

        context.ensure_initialized()

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

        config.set_intra_op_parallelism_threads(10)