Searched refs:assert_no_cpu_op (Results 1 – 2 of 2) sorted by relevance
619 old = theano.config.assert_no_cpu_op623 theano.config.assert_no_cpu_op = 'raise'629 theano.config.assert_no_cpu_op = old634 theano.config.assert_no_cpu_op = 'ignore'638 theano.config.assert_no_cpu_op = old
2438 if config.assert_no_cpu_op == "warn":2441 elif config.assert_no_cpu_op == "raise":2443 elif config.assert_no_cpu_op == "pdb":2448 assert_no_cpu_op = theano.tensor.opt.in2out(local_assert_no_cpu_op, variable2451 optdb.register('gpua_assert_no_cpu_op', assert_no_cpu_op, 49.2,