Searched refs:row_shape_dims (Results 1 – 1 of 1) sorted by relevance
71 std::vector<int64> row_shape_dims(major_dims.begin(), major_dims.end()); in CholeskyUnblocked() local73 row_shape_dims.push_back(1); in CholeskyUnblocked()74 row_shape_dims.push_back(n); in CholeskyUnblocked()77 ShapeUtil::MakeShape(a_shape.element_type(), row_shape_dims)); in CholeskyUnblocked()86 Iota(body_builder, ShapeUtil::MakeShape(S32, row_shape_dims), in CholeskyUnblocked()