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