Searched refs:output_devices (Results 1 – 14 of 14) sorted by relevance
| /external/autotest/client/common_lib/cros/cfm/ |
| D | cras_node_collector_unittest.py | 12 output_devices = (
44 mock_host.run_output.side_effect = [output_devices, output_nodes]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 424 const std::vector<string>& output_devices, const DeviceSet& device_set, in PinArgsAndRets() argument
440 if (output_devices.empty()) { in PinArgsAndRets()
568 DCHECK_GT(output_devices.size(), index); in PinArgsAndRets()
569 VLOG(3) << "Setting output device to " << output_devices[index] in PinArgsAndRets()
571 node->set_assigned_device_name(output_devices[index]); in PinArgsAndRets()
619 if (!options.output_devices.empty() && in ValidateMultiDeviceOptions()
620 options.output_devices.size() != signature.output_arg_size()) { in ValidateMultiDeviceOptions()
624 options.output_devices.size(), in ValidateMultiDeviceOptions()
695 for (const string& device : options.output_devices) { in InstantiateMultiDevice()
746 options.input_devices, options.output_devices, *dev_set, arg_nodes, in InstantiateMultiDevice()
[all …]
|
| D | process_function_library_runtime.h | 153 std::vector<Device*>* output_devices) const;
385 const std::vector<string>& output_devices,
|
| D | process_function_library_runtime_test.cc | 611 if (absl::StrContains(inst_opts.output_devices[0], "GPU")) { in TestTwoDeviceInputOutput()
619 if (absl::StrContains(inst_opts.output_devices[1], "GPU")) { in TestTwoDeviceInputOutput()
639 const std::vector<string>& output_devices) { in MakeOptions() argument
643 inst_opts.output_devices = CompleteDevices(output_devices); in MakeOptions()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | partitioned_function_ops.cc | 159 opts->output_devices.push_back(""); in FillOutputDevices()
161 opts->output_devices.push_back(opts->target); in FillOutputDevices()
|
| D | batch_kernels.cc | 369 opts.output_devices.push_back(cpu_device->name()); in InstantiateFunction()
|
| /external/adhd/cras/client/cras_tests/src/ |
| D | control.rs | 88 for dev in cras_client.output_devices() { in control()
|
| /external/adhd/cras/client/libcras/src/ |
| D | libcras.rs | 329 pub fn output_devices(&self) -> impl Iterator<Item = CrasIodevInfo> { in output_devices() method
330 self.server_state.output_devices() in output_devices()
|
| D | cras_shm.rs | 675 pub fn output_devices(&self) -> impl Iterator<Item = CrasIodevInfo> { in output_devices() method
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 173 options.output_devices.push_back(options.target); in InstantiateFunc()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 665 std::vector<string> output_devices; member
|
| D | function.cc | 1039 for (int i = 0; i < options.output_devices.size(); ++i) { in Canonicalize()
1041 options.output_devices[i], in Canonicalize()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 308 inst_ops.output_devices.resize(num_outputs, device_name); in ConstructFunctionHandle()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 678 inst_opts.output_devices.push_back(inst_opts.target); in Instantiate()
|