Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/mps/
H A DMPSDevice.mm31 if (!_mtl_indexing_library) {
41_mtl_indexing_library = [_mtl_device newLibraryWithSource:[NSString stringWithCString:mps::indexin…
45 …TORCH_CHECK(_mtl_indexing_library, "Failed to create indexing library, error: ", [[error descripti…
47 return _mtl_indexing_library;
72 [_mtl_indexing_library release];
74 _mtl_indexing_library = nil;
77 MPSDevice::MPSDevice() : _mtl_device(nil), _mtl_indexing_library(nil) {
H A DMPSDevice.h76 MTLLibrary_t _mtl_indexing_library; variable