Searched defs:DeviceInstance (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | permuter_test.cc | 140 class DeviceInstance { class in tensorflow::__anon6b05b0d40111::PermuterTest
142 DeviceInstance(int rank, std::vector<int> permutation, DataType dtype, in DeviceInstance() function in tensorflow::__anon6b05b0d40111::PermuterTest::DeviceInstance
|
| D | ring_gatherer_test.cc | 122 class DeviceInstance { class in tensorflow::RingGathererTest
124 DeviceInstance(int rank, int num_subdivs, DataType dtype, in DeviceInstance() function in tensorflow::RingGathererTest::DeviceInstance
|
| D | hierarchical_tree_broadcaster_test.cc | 214 class DeviceInstance { class in tensorflow::__anon11e3a3970111::HierarchicalTreeBroadcasterTest
216 DeviceInstance(int rank, DataType dtype, const TensorShape& shape, in DeviceInstance() function in tensorflow::__anon11e3a3970111::HierarchicalTreeBroadcasterTest::DeviceInstance
|
| D | ring_reducer_test.cc | 157 class DeviceInstance { class in tensorflow::RingReducerTest
159 DeviceInstance(int rank, int num_subdivs, DataType dtype, in DeviceInstance() function in tensorflow::RingReducerTest::DeviceInstance
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 249 class DeviceInstance { class in tensorflow::NcclTestBase
251 DeviceInstance(int rank, const string& device_name, NcclTestBase* parent) in DeviceInstance() function in tensorflow::NcclTestBase::DeviceInstance
|