module DNN

# File lib/dnn.rb, line 19
def self.cudnn_available?
  return false unless defined? ::Cumo
  Cumo::CUDA::CUDNN.available?
end

# File lib/dnn/core/global.rb, line 2
def self.learning_phase
  @learning_phase
end

# File lib/dnn/core/global.rb, line 6
def self.learning_phase=(bool)
  @learning_phase = bool
end

# File lib/dnn.rb, line 24
def self.use_cudnn?
  return false unless ENV["RUBY_DNN_USE_CUDNN"] == "ENABLE"
  cudnn_available?
end

# File lib/dnn.rb, line 15
def self.use_cumo?
  defined? ::Cumo
end