Searched refs:dims_to_remove (Results 1 – 2 of 2) sorted by relevance
196 const std::vector<int64>& dims_to_remove);
1185 const std::vector<int64>& dims_to_remove) { in RemoveShapeDimensions() argument1186 if (sharding.IsTileMaximal() || dims_to_remove.empty()) { in RemoveShapeDimensions()1191 dims_to_remove.size()); in RemoveShapeDimensions()1193 if (absl::c_linear_search(dims_to_remove, i)) { in RemoveShapeDimensions()