Searched refs:dev_set (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | function_cache_test.cc | 223 tensorflow::DeviceSet dev_set; in TEST_P() local
226 for (auto d : device_mgr->ListDevices()) dev_set.AddDevice(d); in TEST_P()
248 kFunctionName, kCpuName, dev_set, eager_ctx, corert, request_ctx_fn, in TEST_P()
260 kFunctionName, kCpuName, dev_set, eager_ctx, corert, request_ctx_fn, in TEST_P()
270 kFunctionName, "", dev_set, eager_ctx, corert, request_ctx_fn, in TEST_P()
|
| D | c_api_tfrt_distributed_impl.cc | 102 void PopulateRemoteDevices(tensorflow::DeviceSet* dev_set) override;
203 tensorflow::DeviceSet* dev_set) { in PopulateRemoteDevices() argument
208 dev_set->AddDevice(device); in PopulateRemoteDevices()
|
| D | c_api_tfrt_distributed_interface.h | 32 virtual void PopulateRemoteDevices(tensorflow::DeviceSet* dev_set) = 0;
|
| D | c_api_tfrt.cc | 1482 tensorflow::DeviceSet dev_set; in Initialize() local
1487 for (auto d : device_mgr->ListDevices()) dev_set.AddDevice(d); in Initialize()
1492 ->PopulateRemoteDevices(&dev_set); in Initialize()
1541 op_name_, device_name_, dev_set, context_->GetEagerContext(), corert, in Initialize()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 839 const std::shared_ptr<DeviceSet> dev_set = device_set(); in InstantiateMultiDevice() local
844 options.input_devices, options.output_devices, *dev_set, arg_nodes, in InstantiateMultiDevice()
874 *dev_set, options.config_proto, &graph, &data->lib_def_, in InstantiateMultiDevice()
898 optimization_options.device_set = dev_set.get(); in InstantiateMultiDevice()
921 dev_set.get(), default_device, in InstantiateMultiDevice()
939 &data->lib_def_, *dev_set, cpu_device, &graph); in InstantiateMultiDevice()
973 PartitionFunctionGraph(*dev_set, std::move(graph), &subgraphs)); in InstantiateMultiDevice()
1064 runner([this, &pair, dev_set, comp_data, unique_name, data_lib_def, in InstantiateMultiDevice()
1069 dev_set->FindDeviceByName(target)->device_type(); in InstantiateMultiDevice()
|