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