Searched defs:actual_dim (Results 1 – 4 of 4) sorted by relevance
59 auto actual_dim = actualDim(dim, /*wrap_dim=*/false); in refreshTensorMetadata() local92 auto actual_dim = actualDim(d, /*wrap_dim=*/ true); in size_custom() local102 auto actual_dim = actualDim(d, /*wrap_dim=*/ true); in sym_size_custom() local
122 auto actual_dim = c10::maybe_wrap_dim(d, logical_dim); in squeeze_dims__batching_rule() local
239 auto actual_dim = c10::maybe_wrap_dim(d, ndim - 1); in squeeze_dims_batch_rule() local
28 auto actual_dim = actualDim(dim, /*wrap_dim=*/false); in BatchedTensorImpl() local55 for (const auto actual_dim : c10::irange(kVmapMaxTensorDims)) { in actualDim() local