Searched defs:has_batch_dim (Results 1 – 1 of 1) sorted by relevance
44 int64_t getPhysicalDim(const Tensor& tensor, bool has_batch_dim, int64_t logical_dim) { in getPhysicalDim()55 VmapDimVector getPhysicalDims(const Tensor& tensor, bool has_batch_dim, IntArrayRef logical_dims) { in getPhysicalDims()