| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_testlib.h | 71 std::vector<DeviceAttributes> devices_; variable
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_placer.h | 50 std::unordered_map<string, DeviceProperties> devices_; variable
|
| D | graph_memory_test.cc | 28 std::unordered_map<string, DeviceProperties> devices_; member in tensorflow::grappler::__anondce669160111::GraphMemoryTest
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer.h | 99 const DeviceSet* const devices_; // Not owned. variable
|
| D | device_set.h | 110 std::vector<Device*> devices_; variable
|
| D | device_mgr.h | 180 std::vector<std::unique_ptr<Device>> devices_; variable
|
| D | device_set_test.cc | 57 DeviceSet devices_; member in tensorflow::__anon0a0fc39b0111::DeviceSetTest
|
| D | direct_session.h | 340 std::vector<Device*> devices_; // not owned variable
|
| D | placer_test.cc | 351 DeviceSet devices_; member in tensorflow::__anonced950570111::PlacerTest
|
| /external/tensorflow/tensorflow/tools/optimization/ |
| D | optimization_pass_runner.h | 53 std::vector<std::unique_ptr<Device>> devices_; variable
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | scheduler.h | 84 const DeviceSet* devices_; variable
|
| D | remote_device_test.cc | 40 std::vector<Device*> devices_; member in tensorflow::RemoteDeviceTest
|
| D | master_session.h | 130 std::unique_ptr<DeviceSet> devices_; variable
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | multi_device_iterator_ops.cc | 463 const std::vector<string> devices_; member in tensorflow::data::__anon2efe57860111::MultiDeviceIterator
589 std::vector<string> devices_; member in tensorflow::data::__anon2efe57860111::MultiDeviceIteratorHandleOp
626 std::vector<string> devices_; member in tensorflow::data::__anon2efe57860111::AnonymousMultiDeviceIteratorOp
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_c_api_client.h | 291 std::vector<PjRtDevice*> devices_; variable
|
| D | tfrt_cpu_pjrt_client.h | 247 std::vector<PjRtDevice*> devices_; variable
|
| D | pjrt_stream_executor_client.h | 355 std::vector<PjRtDevice*> devices_; variable
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | build_xla_ops_pass_test.cc | 46 std::vector<std::unique_ptr<Device>> devices_; member in tensorflow::__anon20cc14a80111::BuildXlaOpsTest
|
| D | mark_for_compilation_pass.cc | 243 DeviceSet devices_; member in tensorflow::__anon8f9055b30111::MarkForCompilationPassImpl::Cluster
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | outfeed_receiver.cc | 201 std::vector<PjRtDevice*> devices_; member in xla::OutfeedReceiverImpl
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.h | 165 std::vector<std::shared_ptr<PjRtDevice>> devices_; variable
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_node.h | 283 std::set<string> devices_; variable
900 std::set<string> devices_; variable
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 445 static std::vector<std::unique_ptr<BaseGPUDevice>>* devices_; member in tensorflow::NcclManagerTest
451 std::vector<std::unique_ptr<BaseGPUDevice>>* NcclManagerTest<Scalar>::devices_ = member in tensorflow::NcclManagerTest<Scalar>
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision.cc | 1127 std::unordered_map<string, DeviceProperties> devices_; member in tensorflow::grappler::__anon63eee5310111::AutoMixedPrecisionImpl
|