[PHI] Add nanmedian output defs (#51358)

* add nanmedian output defs

* remove the multiclass_nms3 momentum

paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc

@@ -65,9 +65,6 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = {
     "less_equal",
     "less_than",
     "merged_adam",
-    "momentum",
-    "multiclass_nms3",
-    "nanmedian",
     "sync_batch_norm_grad",
     "unique",
     "unique_consecutive_flattened_tensor",

paddle/phi/kernels/cpu/nanmedian_kernel.cc

@@ -207,4 +207,6 @@ PD_REGISTER_KERNEL(nanmedian,
                    float,
                    double,
                    int,
-                   int64_t) {}
+                   int64_t) {
+  kernel->OutputAt(1).SetDataType(phi::DataType::INT64);
+}

paddle/phi/kernels/gpu/nanmedian_kernel.cu

@@ -286,4 +286,6 @@ PD_REGISTER_KERNEL(nanmedian,
                    double,
                    int,
                    int64_t,
-                   phi::dtype::float16) {}
+                   phi::dtype::float16) {
+  kernel->OutputAt(1).SetDataType(phi::DataType::INT64);
+}