Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DLegacyVmapTransforms.h121 struct TORCH_API VmapPhysicalView { struct
122 VmapPhysicalView(Tensor&& tensor, std::bitset<kVmapNumLevels> levels) in VmapPhysicalView() argument
133 // physical_view = VmapPhysicalView(tensor=ones(2, 3, 4, 5), levels={1, 3}) argument
156 std::bitset<kVmapNumLevels> levels_;
157 Tensor tensor_;
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyVmapTransforms.h114 struct TORCH_API VmapPhysicalView { struct
115 VmapPhysicalView(Tensor&& tensor, std::bitset<kVmapNumLevels> levels) in VmapPhysicalView() argument
130 // physical_view = VmapPhysicalView(tensor=ones(2, 3, 4, 5), levels={1, 3}) argument
152 std::bitset<kVmapNumLevels> levels_;
153 Tensor tensor_;