Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/optimization/
Doptimization_pass_runner.h56 std::vector<std::unique_ptr<Device>> devices_; variable
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer.h94 const DeviceSet* const devices_; // Not owned. variable
Ddevice_set_test.cc53 DeviceSet devices_; member in tensorflow::__anonf0ab42df0111::DeviceSetTest
Ddirect_session.h317 std::vector<Device*> devices_; // not owned variable
Dplacer_test.cc271 DeviceSet devices_; member in tensorflow::__anon5fafcd9b0111::PlacerTest
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_testlib.h66 std::vector<DeviceAttributes> devices_; variable
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_placer.h47 std::unordered_map<string, DeviceProperties> devices_; variable
Dgraph_memory_test.cc28 std::unordered_map<string, DeviceProperties> devices_; member in tensorflow::grappler::__anond85ec95a0111::GraphMemoryTest
/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.h132 std::unique_ptr<DeviceSet> devices_; variable
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.h264 std::vector<Device*> devices_; variable
/external/tensorflow/tensorflow/core/kernels/data/
Dmulti_device_iterator_ops.cc356 const std::vector<string> devices_; member in tensorflow::data::__anon2db6daca0111::MultiDeviceIterator
482 std::vector<string> devices_; member in tensorflow::data::__anon2db6daca0111::MultiDeviceIteratorHandleOp
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass_test.cc45 std::vector<std::unique_ptr<Device>> devices_; member in tensorflow::__anondad77c6c0111::BuildXlaOpsTest
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc239 static std::vector<std::unique_ptr<BaseGPUDevice>>* devices_; member in tensorflow::NcclManagerTest
245 std::vector<std::unique_ptr<BaseGPUDevice>>* NcclManagerTest<Scalar>::devices_ = member in tensorflow::NcclManagerTest<Scalar>
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.h284 std::set<string> devices_; variable
901 std::set<string> devices_; variable