Searched defs:major_dims (Results 1 – 3 of 3) sorted by relevance
32 auto major_dims = AsInt64Slice(shape.dimensions()) in SliceInMinorDims() local118 auto major_dims = AsInt64Slice(shape.dimensions()) in DynamicSliceInMinorDims() local
60 absl::Span<const int64> major_dims = in GetMatrixDiagonal() local99 absl::Span<const int64> major_dims = in TriangleMask() local
60 auto major_dims = AsInt64Slice(a_shape.dimensions()) in CholeskyUnblocked() local