Searched refs:DEVICE_PLACEMENT_EXPLICIT (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context_test.cc | 81 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
116 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
144 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
165 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
202 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
242 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT, /*async=*/true); in TEST_F()
309 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT, /*async=*/true); in TEST_F()
342 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
390 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
396 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
|
| D | placement_test.cc | 116 InitContext(options, DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
157 InitContext(options, DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
|
| D | execute.cc | 147 case DEVICE_PLACEMENT_EXPLICIT: in CopyInputToExpectedDevice()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config.py | 300 elif device_policy == context.DEVICE_PLACEMENT_EXPLICIT:
344 context.context().device_policy = context.DEVICE_PLACEMENT_EXPLICIT
|
| D | config_test.py | 100 self.assertEqual(context.DEVICE_PLACEMENT_EXPLICIT,
|
| D | ops.py | 6208 if device_policy not in (None, context.DEVICE_PLACEMENT_EXPLICIT,
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | immediate_execution_context.h | 50 DEVICE_PLACEMENT_EXPLICIT = 0, enumerator
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 60 DEVICE_PLACEMENT_EXPLICIT = pywrap_tfe.TFE_DEVICE_PLACEMENT_EXPLICIT variable
|