Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for send_pred_builder (0.13 sec)

  1. tensorflow/compiler/jit/extract_outside_compilation_pass.cc

        Graph* g) {
      NodeDefBuilder send_pred_builder(name, "XlaSendToHost");
      send_pred_builder.Attr("Tinput", DT_BOOL);
      send_pred_builder.Attr("key", absl::StrCat(host_transfer_key, "_dtoh_0"));
      send_pred_builder.Attr(kXlaTokenInputNodesAttrName,
                             std::vector<string>{kXlaTokenArgNodeName});
      send_pred_builder.Attr(kXlaOriginalOutsideCompilationNodeName, name);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 104.7K bytes
    - Viewed (0)
Back to top