Home
last modified time | relevance | path

Searched defs:_AllocationMetadata (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DCUDAPluggableAllocator.h58 struct _AllocationMetadata { struct
64 size_t size;
65 c10::DeviceIndex device_idx;
66 cudaStream_t stream;
H A DCUDAPluggableAllocator.cpp32 _AllocationMetadata::_AllocationMetadata() in _AllocationMetadata() function in torch::cuda::CUDAPluggableAllocator::_AllocationMetadata
35 _AllocationMetadata::_AllocationMetadata( in _AllocationMetadata() function in torch::cuda::CUDAPluggableAllocator::_AllocationMetadata