message: Enable (u)int8 for functional graphs
  Summary: Fixes T34846
  Test Plan: CI
  Reviewers: jakeh, hakons, vladimirm, #tensorflow, simonl, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved
  Maniphest Tasks: T34846
diff: @@ -51,6 +51,11 @@ static bool OpFilter(KernelDef* kdef) {
  AddDtypeToKernelDefConstraint("Tout", DT_RESOURCE, kdef);
  AddDtypeToKernelDefConstraint("Tin", DT_VARIANT, kdef);
  AddDtypeToKernelDefConstraint("Tout", DT_VARIANT, kdef);
  + // Add support for (u)int8.
  + AddDtypeToKernelDefConstraint("Tin", DT_INT8, kdef);
  + AddDtypeToKernelDefConstraint("Tout", DT_INT8, kdef);
  + AddDtypeToKernelDefConstraint("Tin", DT_UINT8, kdef);
  + AddDtypeToKernelDefConstraint("Tout", DT_UINT8, kdef);
  }
  // Support (u)int8 for data feeding to the IPU.
  @@ -60,6 +65,12 @@ static bool OpFilter(KernelDef* kdef) {
  AddDtypeToKernelDefConstraint("DstT", DT_INT8, kdef);
  AddDtypeToKernelDefConstraint("DstT", DT_UINT8, kdef);
  }
  +
  + if (kdef->op() == "_Arg" || kdef->op() == "_Retval") {
  + AddDtypeToKernelDefConstraint("T", DT_INT8, kdef);
  + AddDtypeToKernelDefConstraint("T", DT_UINT8, kdef);
  + }
  +
  if (kdef->op() == "PopDatastreamInfeedDequeue" ||
  kdef->op() == "PopDatastreamOutfeedEnqueue") {
  AddDtypeToKernelDefConstraint("output_types", DT_INT8, kdef);
  