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() function 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() function 130 // physical_view = VmapPhysicalView(tensor=ones(2, 3, 4, 5), levels={1, 3}) argument 152 std::bitset<kVmapNumLevels> levels_; 153 Tensor tensor_;
|