Home
last modified time | relevance | path

Searched defs:MetalTensorImpl (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/
H A DMetalTensorImpl.h11 struct TORCH_API MetalTensorImpl : public OpaqueTensorImpl<OpaqueHandle> { struct
12 MetalTensorImpl( in MetalTensorImpl() function
44 return "MetalTensorImpl"; in tensorimpl_type_name() argument
51 #endif /* MetalTensorImpl_h*/ argument
H A DMetalTensorUtils.h38 using MetalTensorImpl = at::MetalTensorImpl<MetalTensorImplStorage>; in getTensorImplStorage() local
48 using MetalTensorImpl = at::MetalTensorImpl<MetalTensorImplStorage>; in makeTensor() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/mpscnn/
H A DMPSImageUtils.h40 using MetalTensorImpl = at::MetalTensorImpl<MetalTensorImplStorage>; in imageFromTensor() local