Home
last modified time | relevance | path

Searched refs:logical_scalar_tensor (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesHelper.cpp161 static void handleScalarTypePromotion(Tensor& logical_scalar_tensor, Tensor& second) { in handleScalarTypePromotion() argument
162 auto result_type = at::native::result_type(logical_scalar_tensor[0], second); in handleScalarTypePromotion()
163 if (logical_scalar_tensor.scalar_type() != result_type) { in handleScalarTypePromotion()
164 logical_scalar_tensor = logical_scalar_tensor.to(result_type); in handleScalarTypePromotion()