Searched defs:cpu_enabled (Results 1 – 4 of 4) sorted by relevance
120 bool cpu_enabled = at::autocast::is_autocast_enabled(at::kCPU); in currentSpecialization() local
24 bool cpu_enabled = false; member
437 Tensor _autocast_to_reduced_precision(const Tensor& self, bool cuda_enabled, bool cpu_enabled, Scal… in _autocast_to_reduced_precision()460 Tensor _autocast_to_full_precision(const Tensor& self, bool cuda_enabled, bool cpu_enabled) { in _autocast_to_full_precision()
5131 const auto cpu_enabled = p_node->Input(2).toBool(); in __anon5d9c3eb91f202() local5151 const auto cpu_enabled = p_node->Input(2).toBool(); in __anon5d9c3eb91f402() local