Searched refs:pivot_dims (Results 1 – 1 of 1) sorted by relevance
39 std::vector<int64> pivot_dims = batch_dims; in LuDecomposition() local40 pivot_dims.push_back(std::min(m, n)); in LuDecomposition()44 {a_shape, ShapeUtil::MakeShape(S32, pivot_dims), in LuDecomposition()