Searched refs:TF_SetDevice (Results 1 – 8 of 8) sorted by relevance
213 TF_SetDevice(desc, op_device.c_str()); in BinaryOpHelper()299 TF_SetDevice(desc, "/cpu:0"); in Split3Helper()
266 TF_CAPI_EXPORT extern void TF_SetDevice(TF_OperationDescription* desc,
223 TF_SetDevice(desc, "/cpu:0"); in UseHelper()
789 void TF_SetDevice(TF_OperationDescription* desc, const char* device) { in TF_SetDevice() function
142 TF_SetDevice(d, cdevice); in Java_org_tensorflow_GraphOperationBuilder_setDevice()
357 C.TF_SetDevice(cdesc, cdevice)
1101 m.def("TF_SetDevice", TF_SetDevice); in PYBIND11_MODULE()
1845 pywrap_tf_session.TF_SetDevice(op_desc, compat.as_str(node_def.device))