Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 12 of 12 for DeviceSet (0.1 sec)

  1. tensorflow/c/eager/c_api_distributed_test.cc

          : error_node_(error_node), error_device_(error_device) {}
      tensorflow::Status Run(const std::string& function_name,
                             const tensorflow::DeviceSet& device_set,
                             const tensorflow::ConfigProto& config_proto,
                             const FunctionOptions& function_options,
                             std::unique_ptr<tensorflow::Graph>* graph,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 15 09:49:45 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/build_xla_ops_pass.cc

        const string& function_name, const FunctionLibraryDefinition& flib_def) {
      const FunctionDef* func_def = flib_def.Find(function_name);
      TF_RET_CHECK(func_def) << "Could not find " << function_name;
    
      jit::DeviceSet device_set;
    
      for (const NodeDef& ndef : func_def->node_def()) {
        VLOG(3) << ndef.DebugString();
        if (!ndef.device().empty()) {
          TF_ASSIGN_OR_RETURN(jit::DeviceId device_id,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 24.3K bytes
    - Viewed (0)
Back to top