Searched defs:num_dims_to_squeeze (Results 1 – 1 of 1) sorted by relevance
2935 size_t num_dims_to_squeeze = is_all_reduce ? self.dim() : opt_dim.value().size(); local2983 size_t num_dims_to_squeeze = dim.has_value() && !dim.value().empty() ? local