Home
last modified time | relevance | path

Searched defs:devices_ (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_placer.h50 std::unordered_map<string, DeviceProperties> devices_; variable
Dgraph_memory_test.cc28 std::unordered_map<string, DeviceProperties> devices_; member in tensorflow::grappler::__anon5aa299380111::GraphMemoryTest
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_testlib.h71 std::vector<DeviceAttributes> devices_; variable
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer.h99 const DeviceSet* const devices_; // Not owned. variable
Ddevice_set.h110 std::vector<Device*> devices_; variable
Ddevice_mgr.h175 std::vector<std::unique_ptr<Device>> devices_; variable
Ddevice_set_test.cc57 DeviceSet devices_; member in tensorflow::__anon7d5d833d0111::DeviceSetTest
Ddirect_session.h340 std::vector<Device*> devices_; // not owned variable
Dplacer_test.cc339 DeviceSet devices_; member in tensorflow::__anon97448ef90111::PlacerTest
/external/tensorflow/tensorflow/tools/optimization/
Doptimization_pass_runner.h53 std::vector<std::unique_ptr<Device>> devices_; variable
/external/tensorflow/tensorflow/core/distributed_runtime/
Dscheduler.h84 const DeviceSet* devices_; variable
Dremote_device_test.cc40 std::vector<Device*> devices_; member in tensorflow::RemoteDeviceTest
Dmaster_session.h130 std::unique_ptr<DeviceSet> devices_; variable
/external/tensorflow/tensorflow/core/kernels/data/
Dmulti_device_iterator_ops.cc379 const std::vector<string> devices_; member in tensorflow::data::__anon2e5a99280111::MultiDeviceIterator
505 std::vector<string> devices_; member in tensorflow::data::__anon2e5a99280111::MultiDeviceIteratorHandleOp
537 std::vector<string> devices_; member in tensorflow::data::__anon2e5a99280111::AnonymousMultiDeviceIteratorOp
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.h138 std::vector<std::shared_ptr<PjRtDevice>> devices_; variable
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass_test.cc46 std::vector<std::unique_ptr<Device>> devices_; member in tensorflow::__anonfdd1c88a0111::BuildXlaOpsTest
Dmark_for_compilation_pass.cc231 DeviceSet devices_; member in tensorflow::__anonc4d28a150111::MarkForCompilationPassImpl::Cluster
/external/tensorflow/tensorflow/compiler/xla/python/
Doutfeed_receiver.cc204 std::vector<PjRtDevice*> devices_; member in xla::OutfeedReceiverImpl
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.h283 std::set<string> devices_; variable
900 std::set<string> devices_; variable
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.h258 std::vector<PjRtDevice*> devices_; variable
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc445 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>