Home
last modified time | relevance | path

Searched refs:output_devices (Results 1 – 8 of 8) sorted by relevance

/external/autotest/client/common_lib/cros/cfm/
Dcras_node_collector_unittest.py12 output_devices = (
44 mock_host.run_output.side_effect = [output_devices, output_nodes]
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.cc365 const std::vector<string>& output_devices, const DeviceSet& device_set, in PinArgsAndRets() argument
381 if (output_devices.empty()) { in PinArgsAndRets()
490 DCHECK_GT(output_devices.size(), index); in PinArgsAndRets()
491 VLOG(3) << "Setting output device to " << output_devices[index] in PinArgsAndRets()
493 node->set_assigned_device_name(output_devices[index]); in PinArgsAndRets()
541 if (!options.output_devices.empty() && in ValidateMultiDeviceOptions()
542 options.output_devices.size() != signature.output_arg_size()) { in ValidateMultiDeviceOptions()
546 options.output_devices.size(), in ValidateMultiDeviceOptions()
617 for (const string& device : options.output_devices) { in InstantiateMultiDevice()
667 options.input_devices, options.output_devices, device_set_, arg_nodes, in InstantiateMultiDevice()
[all …]
Dprocess_function_library_runtime.h147 std::vector<Device*>* output_devices) const;
375 const std::vector<string>& output_devices,
Dprocess_function_library_runtime_test.cc554 if (absl::StrContains(inst_opts.output_devices[0], "GPU")) { in TestTwoDeviceInputOutput()
562 if (absl::StrContains(inst_opts.output_devices[1], "GPU")) { in TestTwoDeviceInputOutput()
582 const std::vector<string>& output_devices) { in MakeOptions() argument
586 inst_opts.output_devices = CompleteDevices(output_devices); in MakeOptions()
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc153 opts->output_devices.push_back(""); in FillOutputDevices()
155 opts->output_devices.push_back(opts->target); in FillOutputDevices()
/external/tensorflow/tensorflow/core/framework/
Dfunction.h573 std::vector<string> output_devices; member
Dfunction.cc1014 for (int i = 0; i < options.output_devices.size(); ++i) { in Canonicalize()
1016 options.output_devices[i], in Canonicalize()
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc591 inst_opts.output_devices.push_back(inst_opts.target); in Instantiate()