Searched defs:major_dims (Results 1 – 4 of 4) sorted by relevance
52 auto major_dims = AsInt64Slice(shape.dimensions()) in SliceInMinorDims() local139 auto major_dims = AsInt64Slice(shape.dimensions()) in DynamicSliceInMinorDims() local
65 absl::Span<const int64> major_dims = in GetDiagonalMask() local219 absl::Span<const int64> major_dims = in TriangleMask() local
65 auto major_dims = AsInt64Slice(a_shape.dimensions()) in CholeskyUnblocked() local
295 int64_t major_dims = type.getRank() - minor_starts.size(); in SliceInMinorDims() local334 int64_t major_dims = type.getRank() - minor_starts.size(); in DynamicSliceInMinorDims() local