HomeSort by relevance Sort by last modified time
    Searched refs:TF_SetDevice (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/c/
c_test_util.cc 172 TF_SetDevice(desc, op_device.c_str());
249 // they are usually available on GPUs. After doing this, remove TF_SetDevice
251 TF_SetDevice(desc, "/cpu:0");
c_api.h 432 TF_CAPI_EXPORT extern void TF_SetDevice(TF_OperationDescription* desc,
    [all...]
c_api_function_test.cc 220 TF_SetDevice(desc, "/cpu:0");
    [all...]
c_api.cc     [all...]
  /external/tensorflow/tensorflow/java/src/main/native/
operation_builder_jni.cc 140 TF_SetDevice(d, cdevice);

Completed in 408 milliseconds