Searched defs:logical_dims (Results 1 – 2 of 2) sorted by relevance
55 VmapDimVector getPhysicalDims(const Tensor& tensor, bool has_batch_dim, IntArrayRef logical_dims) { in getPhysicalDims()
65 auto logical_dims = opt_logical_dims.value(); in getPhysicalDims() local