Searched defs:device_set_ (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/clusters/ | ||
| D | virtual_cluster.h | 55 const DeviceSet* device_set_ = nullptr; variable |
| D | single_machine.h | 78 std::unique_ptr<DeviceSet> device_set_; variable |
| /external/tensorflow/tensorflow/core/common_runtime/ | ||
| D | colocation_graph.h | 244 const DeviceSet* device_set_; // Not owned. variable |
| D | graph_execution_state.h | 195 const DeviceSet* device_set_; // Not owned variable |
| D | partitioning_utils_test.cc | 108 DeviceSet device_set_; member in tensorflow::__anone4792f6c0111::PartitioningUtilsTest |
| D | direct_session.h | 318 DeviceSet device_set_; variable |
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ | ||
| D | convert_graph_test.cc | 147 const DeviceSet* device_set_; member in tensorflow::tensorrt::convert::FakeCluster |