Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for recv_at_host_builder (0.3 sec)

  1. tensorflow/compiler/jit/extract_outside_compilation_pass.cc

      device_ordinal_value.set_placeholder("_device_ordinal");
      recv_at_host_builder.Attr("device_ordinal", device_ordinal_value);
      recv_at_host_builder.Attr(
          "key", absl::StrCat("host_compute_channel_", oc_cluster_name));
      recv_at_host_builder.Attr(kXlaHasHostTransferAttrName, true);
      recv_at_host_builder.Input(key_placeholder->name(), 0, DT_STRING);
      TF_RETURN_IF_ERROR(recv_at_host_builder.Finalize(&recv_at_host_def));
    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