Searched refs:logical_dims (Results 1 – 6 of 6) sorted by relevance
55 VmapDimVector getPhysicalDims(const Tensor& tensor, bool has_batch_dim, IntArrayRef logical_dims) { in getPhysicalDims() argument60 result.reserve(logical_dims.size()); in getPhysicalDims()61 for (auto d : logical_dims){ in getPhysicalDims()
53 VmapDimVector VmapPhysicalView::getPhysicalDims(IntArrayRef logical_dims) const { in getPhysicalDims()58 for (auto dim : logical_dims) { in getPhysicalDims()
139 VmapDimVector getPhysicalDims(IntArrayRef logical_dims) const;
38 VmapDimVector getPhysicalDims(const Tensor& tensor, bool has_batch_dim, IntArrayRef logical_dims);
65 auto logical_dims = opt_logical_dims.value(); in getPhysicalDims() local66 for (auto dim : logical_dims) { in getPhysicalDims()
136 VmapDimVector getPhysicalDims(OptionalIntArrayRef logical_dims) const;