Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for BuildXlaHostComputeNodeDef (0.19 sec)

  1. tensorflow/compiler/jit/extract_outside_compilation_pass.cc

      return absl::StrCat("outside_compilation_", original_oc_name,
                          "_host_compute");
    }
    
    // Builds XlaHostCompute NodeDef from the outside compilation call node.
    absl::StatusOr<NodeDef> BuildXlaHostComputeNodeDef(
        const Node* call_node, const std::map<string, int>& host_compute_core,
        const absl::flat_hash_map<string, std::vector<string>>& cluster_deps) {
      string original_oc_name;
      TF_RETURN_IF_ERROR(GetNodeAttr(
    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