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

  /external/tensorflow/tensorflow/c/
c_test_util.cc 213 TF_SetDevice(desc, op_device.c_str());
297 // they are usually available on GPUs. After doing this, remove TF_SetDevice
299 TF_SetDevice(desc, "/cpu:0");
c_api_function_test.cc 223 TF_SetDevice(desc, "/cpu:0");
    [all...]
c_api.h 466 TF_CAPI_EXPORT extern void TF_SetDevice(TF_OperationDescription* desc,
    [all...]
c_api.cc     [all...]
  /external/tensorflow/tensorflow/go/
graph.go 334 C.TF_SetDevice(cdesc, cdevice)
  /external/tensorflow/tensorflow/java/src/main/native/
operation_builder_jni.cc 140 TF_SetDevice(d, cdevice);
  /external/tensorflow/tensorflow/python/framework/
ops.py     [all...]

Completed in 133 milliseconds