Searched defs:device_set (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 131 void SetDeviceSet(const DeviceSet* device_set) { device_set_ = device_set; } in SetDeviceSet()
191 DeviceSet device_set; in TEST() local
|
| D | convert_graph.cc | 760 auto device_set = params.cluster->GetDeviceSet(); in GetDeviceAndAllocator() local
|
| /external/tensorflow/tensorflow/tools/optimization/ |
| D | optimization_pass_runner.cc | 125 DeviceSet device_set; in RunMain() local
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | virtual_cluster.cc | 46 VirtualCluster::VirtualCluster(const DeviceSet* device_set) in VirtualCluster()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.h | 42 const DeviceSet* device_set = nullptr; member
|
| D | optimization_registry.h | 49 const DeviceSet* device_set = nullptr; // Not owned. member
|
| D | partitioning_utils.cc | 26 const DeviceSet& device_set, std::unique_ptr<Graph> graph, in PartitionFunctionGraph()
|
| D | process_function_library_runtime.cc | 305 const std::vector<string>& output_devices, const DeviceSet& device_set, in PinArgsAndRets()
568 DeviceSet device_set; in InstantiateMultiDevice() local
|
| D | graph_execution_state.cc | 292 Status LookupDevice(const DeviceSet& device_set, const string& tensor_name, in LookupDevice()
|
| D | colocation_graph.cc | 453 const DeviceSet* device_set, in ColocationGraph()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client_library.cc | 83 se::Platform* platform, const absl::optional<std::set<int>>& device_set) { in GetOrCreateLocalClient()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 262 std::vector<string> filtered_worker_list) { in Init()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 716 FunctionLibraryDefinition* flib, const DeviceSet& device_set, in OptimizeGraph()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | increase_dynamism_for_auto_jit_pass_test.cc | 68 std::unique_ptr<DeviceSet> device_set(new DeviceSet()); in IncreaseDynamismForAutoJit() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master.cc | 372 std::unique_ptr<DeviceSet> device_set; in CreateSession() local
|
| D | master_session.cc | 1174 std::unique_ptr<DeviceSet> device_set, in MasterSession()
|