Searched refs:ContextDevicePlacementPolicy (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | immediate_execution_context.h | 47 enum ContextDevicePlacementPolicy { enum
138 ContextDevicePlacementPolicy policy) = 0;
140 virtual ContextDevicePlacementPolicy GetDevicePlacementPolicy() const = 0;
|
| D | c_api.cc | 120 static_cast<tensorflow::ContextDevicePlacementPolicy>( in TFE_NewContext()
146 static_cast<tensorflow::ContextDevicePlacementPolicy>( in TFE_NewContext()
254 static_cast<tensorflow::ContextDevicePlacementPolicy>(policy)); in TFE_ContextSetThreadLocalDevicePlacementPolicy()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle_test.cc | 41 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, false, in TEST()
107 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, in PackedTensorHandleTest()
257 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, false, in TEST()
289 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, false, in TEST()
343 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, in RemoteTensorHandleTest()
382 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, in TEST_F()
417 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, false, in TEST()
|
| D | context.h | 99 ContextDevicePlacementPolicy default_device_placement_policy,
173 ContextDevicePlacementPolicy policy) override;
176 ContextDevicePlacementPolicy GetDevicePlacementPolicy() const override;
572 const ContextDevicePlacementPolicy default_device_placement_policy_;
577 std::unordered_map<std::thread::id, ContextDevicePlacementPolicy>
|
| D | eager_operation_test.cc | 30 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, false, in TEST()
|
| D | custom_device_test.cc | 91 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, false, in TEST()
116 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, false, in TEST()
|
| D | eager_op_rewrite_registry_test.cc | 50 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, false, in TEST()
|
| D | execute_node_test.cc | 70 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, false, in TEST()
|
| D | placement_test.cc | 83 ContextDevicePlacementPolicy policy) { in InitContext()
|
| D | mkl_eager_op_rewrite_test.cc | 45 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, in CreateOp()
|
| D | context_test.cc | 58 ContextDevicePlacementPolicy policy) { in InitContext()
|
| D | context.cc | 78 ContextDevicePlacementPolicy default_device_placement_policy, bool async, in EagerContext()
396 ContextDevicePlacementPolicy policy) { in SetThreadLocalDevicePlacementPolicy()
401 ContextDevicePlacementPolicy EagerContext::GetDevicePlacementPolicy() const { in GetDevicePlacementPolicy()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | delegate_data.cc | 48 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, in Prepare()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | test_utils.cc | 47 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, in CreateTestingEagerContext()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_mgr_test.cc | 56 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, in RemoteMgrTest()
|
| D | eager_service_impl.cc | 276 opts, tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, in CreateContext()
|
| D | eager_service_impl_test.cc | 1219 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, in TEST_F()
|