Home
last modified time | relevance | path

Searched refs:_USE_C_API (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/python/client/
Dsession_list_devices_test.py92 self.prev_use_c_api = ops._USE_C_API
93 ops._USE_C_API = False
97 ops._USE_C_API = self.prev_use_c_api
106 self.prev_use_c_api = ops._USE_C_API
107 ops._USE_C_API = True
111 ops._USE_C_API = self.prev_use_c_api
Dsession_partial_run_test.py290 self.prev_use_c_api = ops._USE_C_API
291 ops._USE_C_API = False
295 ops._USE_C_API = self.prev_use_c_api
304 self.prev_use_c_api = ops._USE_C_API
305 ops._USE_C_API = True
309 ops._USE_C_API = self.prev_use_c_api
Dsession_test.py171 if ops._USE_C_API:
1508 if ops._USE_C_API:
1821 self._original_use_c_api_value = ops._USE_C_API
1822 ops._USE_C_API = True
1826 ops._USE_C_API = self._original_use_c_api_value
Dsession_clusterspec_prop_test.py45 ops._USE_C_API = True
Dsession.py625 self._created_with_new_api = ops._USE_C_API
/external/tensorflow/tensorflow/python/framework/
Dimporter_test.py377 if ops._USE_C_API:
428 if ops._USE_C_API:
443 if ops._USE_C_API:
460 if ops._USE_C_API:
484 if ops._USE_C_API:
499 if ops._USE_C_API:
512 if ops._USE_C_API:
525 if ops._USE_C_API:
538 if ops._USE_C_API:
552 if ops._USE_C_API:
[all …]
Dops_test.py445 if ops._USE_C_API:
464 if ops._USE_C_API: return
478 if not ops._USE_C_API: return
516 if not ops._USE_C_API: return
587 if ops._USE_C_API:
603 if not ops._USE_C_API:
618 if ops._USE_C_API: return
627 if not ops._USE_C_API: return
644 if ops._USE_C_API:
745 if ops._USE_C_API:
[all …]
Dtest_util.py320 prev_value = ops._USE_C_API
321 ops._USE_C_API = use_c_api
330 ops._USE_C_API = prev_value
338 return ops._USE_C_API and IsGoogleCudaEnabled()
Dsubscribe.py140 if ops._USE_C_API: # pylint: disable=protected-access
Dops.py65 _USE_C_API = os.getenv("TF_C_API_GRAPH_CONSTRUCTION", "0") is not "0" variable
371 if _USE_C_API:
469 if not _USE_C_API:
2764 if _USE_C_API or self._use_c_api_hack():
3033 if _USE_C_API:
Dfunction_test.py788 if ops._USE_C_API: # pylint: disable=protected-access
1512 ops._USE_C_API = True
1526 ops._USE_C_API = False
Dfunction.py823 if ops._USE_C_API:
/external/tensorflow/tensorflow/contrib/framework/python/framework/
Dtensor_util_test.py218 if ops._USE_C_API:
230 if ops._USE_C_API:
/external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/bijectors/
Dreshape_test.py307 if ops._USE_C_API:
314 if ops._USE_C_API:
/external/tensorflow/tensorflow/contrib/training/python/training/
Dbatch_sequences_with_states_test.py511 self._prev_value = ops._USE_C_API
512 ops._USE_C_API = True
517 ops._USE_C_API = self._prev_value
/external/tensorflow/tensorflow/python/kernel_tests/
Darray_ops_test.py1032 if ops._USE_C_API:
1044 if ops._USE_C_API:
1065 if ops._USE_C_API:
/external/tensorflow/tensorflow/python/saved_model/
Dsaved_model_test.py60 if ops._USE_C_API:
983 if ops._USE_C_API:
/external/tensorflow/tensorflow/python/ops/
Dgradients_test.py288 if not ops._USE_C_API: return
/external/tensorflow/tensorflow/contrib/losses/python/losses/
Dloss_ops_test.py476 if ops._USE_C_API:
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_test.py1328 if ops._USE_C_API:
1865 if ops._USE_C_API: