Searched defs:numBatchDims (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ | ||
| H A D | LegacyVmapTransforms.cpp | 45 int64_t VmapPhysicalView::numBatchDims() const { in numBatchDims() function in at::functorch::VmapPhysicalView |
| /aosp_15_r20/external/pytorch/aten/src/ATen/ | ||
| H A D | LegacyVmapTransforms.cpp | 51 int64_t VmapPhysicalView::numBatchDims() const { in numBatchDims() function in at::VmapPhysicalView |