TfLiteRegistration* Register_MATRIX_SET_DIAG() {
  static TfLiteRegistration r = {nullptr, nullptr, matrix_set_diag::Prepare,
                                 matrix_set_diag::Eval};
  return &r;
}
TfLiteRegistration* Register_MATRIX_SET_DIAG() {
  static TfLiteRegistration r = {nullptr, nullptr, matrix_set_diag::Prepare,
                                 matrix_set_diag::Eval};
  return &r;
}