Searched defs:getPhysicalDim (Results 1 – 3 of 3) sorted by relevance
44 int64_t getPhysicalDim(const Tensor& tensor, bool has_batch_dim, int64_t logical_dim) { in getPhysicalDim() function
64 int64_t VmapPhysicalView::getPhysicalDim(int64_t logical_dim) const { in getPhysicalDim() function in at::functorch::VmapPhysicalView
77 int64_t VmapPhysicalView::getPhysicalDim(int64_t logical_dim) const { in getPhysicalDim() function in at::VmapPhysicalView