Searched refs:col_shape_dims (Results 1 – 1 of 1) sorted by relevance
72 std::vector<int64> col_shape_dims(major_dims.begin(), major_dims.end()); in CholeskyUnblocked() local79 col_shape_dims.push_back(n); in CholeskyUnblocked()80 col_shape_dims.push_back(1); in CholeskyUnblocked()83 ShapeUtil::MakeShape(a_shape.element_type(), col_shape_dims)); in CholeskyUnblocked()89 Iota(body_builder, ShapeUtil::MakeShape(S32, col_shape_dims), in CholeskyUnblocked()