Searched defs:VmapPhysicalView (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | LegacyVmapTransforms.h | 121 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 D | LegacyVmapTransforms.h | 114 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_;
|