Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DLegacyVmapTransforms.h164 struct TORCH_API VmapPhysicalToLogicalMap { struct
165 VmapPhysicalToLogicalMap(std::bitset<kVmapNumLevels> levels): levels_(levels) {} in VmapPhysicalToLogicalMap() function
183 std::bitset<kVmapNumLevels> levels_;
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyVmapTransforms.h160 struct TORCH_API VmapPhysicalToLogicalMap { struct
161 VmapPhysicalToLogicalMap(std::bitset<kVmapNumLevels> levels) in VmapPhysicalToLogicalMap() argument
180 std::bitset<kVmapNumLevels> levels_;