Searched refs:input_devices (Results 1 – 12 of 12) sorted by relevance
| /external/autotest/client/common_lib/cros/cfm/ |
| D | cras_node_collector_unittest.py | 57 input_devices = (
85 mock_host.run_output.side_effect = [input_devices, input_nodes]
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 233 std::vector<Device*> input_devices, in KernelAndDeviceFunc() argument
245 input_devices_(std::move(input_devices)), in KernelAndDeviceFunc()
|
| D | kernel_and_device.cc | 145 options.input_devices.push_back(device->name()); in InstantiateFunc()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | partitioned_function_ops.cc | 213 opts.input_devices.push_back(handle.device()); in Instantiate()
215 opts.input_devices.push_back(opts.target); in Instantiate()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime_test.cc | 542 if (absl::StrContains(inst_opts.input_devices[0], "GPU")) { in TestTwoDeviceInputOutput()
546 if (absl::StrContains(inst_opts.input_devices[1], "GPU")) { in TestTwoDeviceInputOutput()
581 const string& target, const std::vector<string>& input_devices, in MakeOptions() argument
585 inst_opts.input_devices = CompleteDevices(input_devices); in MakeOptions()
|
| D | process_function_library_runtime.cc | 364 const std::vector<string>& input_devices, in PinArgsAndRets() argument
377 node->set_assigned_device_name(input_devices[index]); in PinArgsAndRets()
534 if (options.input_devices.size() != signature.input_arg_size()) { in ValidateMultiDeviceOptions()
538 options.input_devices.size(), in ValidateMultiDeviceOptions()
612 for (const string& device : options.input_devices) { in InstantiateMultiDevice()
667 options.input_devices, options.output_devices, device_set_, arg_nodes, in InstantiateMultiDevice()
|
| D | process_function_library_runtime.h | 374 Status PinArgsAndRets(const std::vector<string>& input_devices,
|
| D | function.cc | 1510 std::vector<string> input_devices(caller.in_edges().size()); in InputDevices() local
1518 input_devices[edge->dst_input()] = input_device; in InputDevices()
1525 for (int i = 0; i < input_devices.size(); ++i) { in InputDevices()
1528 << " device: " << input_devices[i] in InputDevices()
1533 return input_devices; in InputDevices()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 560 inst_opts.input_devices.push_back(inst_opts.target); in Instantiate()
574 inst_opts.input_devices.push_back(handle.device()); in Instantiate()
583 inst_opts.input_devices.push_back(cpu_device->name()); in Instantiate()
586 inst_opts.input_devices.push_back(inst_opts.target); in Instantiate()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 545 std::vector<string> input_devices; member
|
| D | function.cc | 1000 options.input_devices.size()); in Canonicalize()
1010 for (int i = 0; i < options.input_devices.size(); ++i) { in Canonicalize()
1011 entries.push_back(AttrKeyAndValue("_input_dev", i, options.input_devices[i], in Canonicalize()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 560 options.input_devices.push_back(local_device_); in TEST_F()
|