Searched refs:major_dims (Results 1 – 4 of 4) sorted by relevance
65 auto major_dims = AsInt64Slice(a_shape.dimensions()) in CholeskyUnblocked() local80 std::vector<int64> row_shape_dims(major_dims.begin(), major_dims.end()); in CholeskyUnblocked()81 std::vector<int64> col_shape_dims(major_dims.begin(), major_dims.end()); in CholeskyUnblocked()
52 auto major_dims = AsInt64Slice(shape.dimensions()) in SliceInMinorDims() local60 padded_start.begin() + major_dims.size()); in SliceInMinorDims()64 std::copy(major_dims.begin(), major_dims.end(), padded_end.begin()); in SliceInMinorDims()65 std::copy(end.begin(), end.end(), padded_end.begin() + major_dims.size()); in SliceInMinorDims()139 auto major_dims = AsInt64Slice(shape.dimensions()) in DynamicSliceInMinorDims() local144 auto padded_sizes = ConcatVectors(major_dims, sizes); in DynamicSliceInMinorDims()
65 absl::Span<const int64> major_dims = in GetDiagonalMask() local70 auto mask = Broadcast(indicator, major_dims); in GetDiagonalMask()219 absl::Span<const int64> major_dims = in TriangleMask() local225 return Broadcast(indicator, major_dims); in TriangleMask()
295 int64_t major_dims = type.getRank() - minor_starts.size(); in SliceInMinorDims() local297 slice_starts.begin() + major_dims); in SliceInMinorDims()300 slice_limits.begin() + major_dims); in SliceInMinorDims()334 int64_t major_dims = type.getRank() - minor_starts.size(); in DynamicSliceInMinorDims() local337 slice_sizes.begin() + major_dims); in DynamicSliceInMinorDims()