Searched defs:cuda_enabled (Results 1 – 3 of 3) sorted by relevance
5 inline constexpr bool cuda_enabled() { in cuda_enabled() function
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()
5130 const auto cuda_enabled = p_node->Input(1).toBool(); in __anon5d9c3eb91f202() local5150 const auto cuda_enabled = p_node->Input(1).toBool(); in __anon5d9c3eb91f402() local