Searched defs:major_dims (Results 1 – 4 of 4) sorted by relevance
53 auto major_dims = shape.dimensions().subspan( in SliceInMinorDims() local140 auto major_dims = shape.dimensions().subspan( in DynamicSliceInMinorDims() local
66 absl::Span<const int64_t> major_dims = in GetDiagonalMask() local220 absl::Span<const int64_t> major_dims = in TriangleMask() local
65 auto major_dims = a_shape.dimensions().subspan( in CholeskyUnblocked() local
226 int64_t major_dims = type.getRank() - minor_starts.size(); in SliceInMinorDims() local265 int64_t major_dims = type.getRank() - minor_starts.size(); in DynamicSliceInMinorDims() local