Home
last modified time | relevance | path

Searched defs:getPhysicalDim (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesHelper.cpp44 int64_t getPhysicalDim(const Tensor& tensor, bool has_batch_dim, int64_t logical_dim) { in getPhysicalDim() function
H A DLegacyVmapTransforms.cpp64 int64_t VmapPhysicalView::getPhysicalDim(int64_t logical_dim) const { in getPhysicalDim() function in at::functorch::VmapPhysicalView
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyVmapTransforms.cpp77 int64_t VmapPhysicalView::getPhysicalDim(int64_t logical_dim) const { in getPhysicalDim() function in at::VmapPhysicalView