Searched defs:dim_set (Results 1 – 2 of 2) sorted by relevance
546 const absl::flat_hash_set<int64_t> dim_set(dims.begin(), dims.end()); in Reduce4DTo1D() local
626 absl::flat_hash_set<int64_t> dim_set; in ValidateDotDimensionNumbers() local