Searched refs:dims_to_keep (Results 1 – 1 of 1) sorted by relevance
165 std::vector<int64> dims_to_keep; in IsReductionToVector() local168 dims_to_keep.push_back(dim); in IsReductionToVector()172 dims_to_keep) && in IsReductionToVector()176 [&](int64 dim) { return absl::c_count(dims_to_keep, dim); }, in IsReductionToVector()