Searched defs:Autocast (Results 1 – 6 of 6) sorted by relevance
57 TORCH_LIBRARY_IMPL(aten, Autocast, m) { in TORCH_LIBRARY_IMPL() argument
165 TORCH_LIBRARY_IMPL(_, Autocast, m) { in TORCH_LIBRARY_IMPL() argument169 TORCH_LIBRARY_IMPL(aten, Autocast, m) { in TORCH_LIBRARY_IMPL() argument
122 TORCH_LIBRARY_IMPL(aten, Autocast, m) { in TORCH_LIBRARY_IMPL() argument
521 void Autocast(const std::shared_ptr<Graph>& graph) { in Autocast() function
496 Autocast = AutocastCUDA, enumerator
112 Autocast = auto() variable in DispatchKey