Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_SetDevice (0.06 sec)

  1. 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 Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  2. 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 Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
Back to top