Searched refs:arg_dims (Results 1 – 1 of 1) sorted by relevance
2992 std::vector<int64> arg_dims = arg->dimensions(); in HandleReduce() local2993 absl::c_sort(arg_dims); in HandleReduce()2997 for (int64 arg_dim : arg_dims) { in HandleReduce()3007 std::merge(arg_dims.begin(), arg_dims.end(), reduce_dims.begin(), in HandleReduce()