Searched defs:cpu_dtype (Results 1 – 2 of 2) sorted by relevance
437 …t Tensor& self, bool cuda_enabled, bool cpu_enabled, ScalarType cuda_dtype, ScalarType cpu_dtype) { in _autocast_to_reduced_precision()
5133 const auto cpu_dtype = p_node->Input(4).toScalarType(); in __anon7ebb59391f202() local