Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TF_SetDevice (0.2 sec)

  1. tensorflow/c/c_api.h

    TF_CAPI_EXPORT extern TF_OperationDescription* TF_NewOperation(
        TF_Graph* graph, const char* op_type, const char* oper_name);
    
    // Specify the device for `desc`.  Defaults to empty, meaning unconstrained.
    TF_CAPI_EXPORT extern void TF_SetDevice(TF_OperationDescription* desc,
                                            const char* device);
    
    // The calls to TF_AddInput and TF_AddInputList must match (in number,
    Registered: Tue Sep 09 12:39:10 UTC 2025
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  2. tensorflow/c/c_api.cc

                                             const char* oper_name) {
      mutex_lock l(graph->mu);
      return TF_NewOperationLocked(graph, op_type, oper_name);
    }
    
    void TF_SetDevice(TF_OperationDescription* desc, const char* device) {
      desc->node_builder.Device(device);
    }
    
    void TF_AddInput(TF_OperationDescription* desc, TF_Output input) {
    Registered: Tue Sep 09 12:39:10 UTC 2025
    - Last Modified: Tue May 13 06:30:43 UTC 2025
    - 102.3K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_test.cc

        TF_TString_Copy(&data[i], source.c_str(), source.length());
      }
    
      TF_DeleteTensor(t);
    }
    
    }  // namespace
    }  // namespace tensorflow
    
    // TODO(josh11b): Test:
    // * TF_SetDevice(desc, "/job:worker");
    // * control inputs / outputs
    // * targets
    Registered: Tue Sep 09 12:39:10 UTC 2025
    - Last Modified: Fri Dec 27 12:18:10 UTC 2024
    - 97K bytes
    - Viewed (0)
Back to top