Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/mps/operations/
DFastFourierTransform.mm91 TORCH_CHECK(supportsComplex(), "FFT operations are only supported on MacOS 14+");
132 TORCH_CHECK(supportsComplex(), "FFT operations are only supported on MacOS 14+");
158 TORCH_CHECK(supportsComplex(), "FFT operations are only supported on MacOS 14+");
DBinaryOps.mm78 TORCH_CHECK_TYPE(!isComplexType(self.scalar_type()) || mps::supportsComplex(),
387 …if (!mps::supportsComplex() && (c10::isComplexType(self.scalar_type()) || c10::isComplexType(other…
418 !mps::supportsComplex()) {
431 !mps::supportsComplex()) {
DCopy.mm63 TORCH_CHECK(supportsComplex(), "MPS complex tensors conjugation needs MacOS14+");
279 if (dst_.is_complex() && !supportsComplex()) {
DUnaryOps.mm257 if (supportsComplex() || !self.is_complex()) {
502 if (!mps::supportsComplex()) {
/external/pytorch/aten/src/ATen/native/mps/
DOperationUtils.h434 inline bool supportsComplex() { in supportsComplex() function
449 return supportsComplex(); in supportedFloatingOrComplexType()
DOperationUtils.mm55 …TORCH_CHECK_TYPE(supportsComplex(), "MPS complex types are only supported on MacOS 14.0 or newer."…