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)
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)
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)