Searched defs:getPhysicalDims (Results 1 – 3 of 3) sorted by relevance
55 VmapDimVector getPhysicalDims(const Tensor& tensor, bool has_batch_dim, IntArrayRef logical_dims) { in getPhysicalDims() function
53 VmapDimVector VmapPhysicalView::getPhysicalDims(IntArrayRef logical_dims) const { in getPhysicalDims() function in at::functorch::VmapPhysicalView
59 VmapDimVector VmapPhysicalView::getPhysicalDims(OptionalIntArrayRef opt_logical_dims) const { in getPhysicalDims() function in at::VmapPhysicalView