[phi] Add register of yolo_loss (#51316)

* add register of yolo_loss_kernel.cc

* fix something

* Update paddle/phi/kernels/cpu/yolo_loss_kernel.cc

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

@@ -97,8 +97,7 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = {
     "unique",
     "unique_consecutive_flattened_tensor",
     "unique_raw",
-    "viterbi_devode",
-    "yolo_loss"};
+    "viterbi_devode"};
 
 // These Ops can use InferMeta to infer the output dtype
 static std::set<std::string> OpsWithAvailablePhiInferMeta = {

paddle/phi/kernels/cpu/yolo_loss_kernel.cc

@@ -371,4 +371,6 @@ void YoloLossKernel(const Context& dev_ctx,
 }  // namespace phi
 
 PD_REGISTER_KERNEL(
-    yolo_loss, CPU, ALL_LAYOUT, phi::YoloLossKernel, float, double) {}
+    yolo_loss, CPU, ALL_LAYOUT, phi::YoloLossKernel, float, double) {
+  kernel->OutputAt(2).SetDataType(phi::DataType::INT32);
+}