Searched refs:major_dims (Results 1 – 3 of 3) sorted by relevance
32 auto major_dims = AsInt64Slice(shape.dimensions()) in SliceInMinorDims() local40 padded_start.begin() + major_dims.size()); in SliceInMinorDims()44 std::copy(major_dims.begin(), major_dims.end(), padded_end.begin()); in SliceInMinorDims()45 std::copy(end.begin(), end.end(), padded_end.begin() + major_dims.size()); in SliceInMinorDims()118 auto major_dims = AsInt64Slice(shape.dimensions()) in DynamicSliceInMinorDims() local123 auto padded_sizes = ConcatVectors(major_dims, sizes); in DynamicSliceInMinorDims()
60 absl::Span<const int64> major_dims = in GetMatrixDiagonal() local65 auto mask = Broadcast(indicator, major_dims); in GetMatrixDiagonal()99 absl::Span<const int64> major_dims = in TriangleMask() local105 return Broadcast(indicator, major_dims); in TriangleMask()
60 auto major_dims = AsInt64Slice(a_shape.dimensions()) in CholeskyUnblocked() local71 std::vector<int64> row_shape_dims(major_dims.begin(), major_dims.end()); in CholeskyUnblocked()72 std::vector<int64> col_shape_dims(major_dims.begin(), major_dims.end()); in CholeskyUnblocked()