Searched refs:output_devices (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 305 const std::vector<string>& output_devices, const DeviceSet& device_set, in PinArgsAndRets() argument
323 if (output_devices.empty()) { in PinArgsAndRets()
398 DCHECK_GT(output_devices.size(), index); in PinArgsAndRets()
399 VLOG(3) << "Setting output device to " << output_devices[index] in PinArgsAndRets()
401 node->set_assigned_device_name(output_devices[index]); in PinArgsAndRets()
452 if (!options.output_devices.empty() && in ValidateMultiDeviceOptions()
453 options.output_devices.size() != signature.output_arg_size()) { in ValidateMultiDeviceOptions()
458 options.output_devices.size(), in ValidateMultiDeviceOptions()
537 for (const string& device : options.output_devices) { in InstantiateMultiDevice()
574 options.input_devices, options.output_devices, device_set, graph.get())); in InstantiateMultiDevice()
[all …]
|
| D | process_function_library_runtime_test.cc | 482 if (str_util::StrContains(inst_opts.output_devices[0], "GPU")) { in TestTwoDeviceInputOutput()
490 if (str_util::StrContains(inst_opts.output_devices[1], "GPU")) { in TestTwoDeviceInputOutput()
512 const std::vector<string>& output_devices) { in MakeOptions() argument
516 inst_opts.output_devices = CompleteDevices(output_devices); in MakeOptions()
|
| D | process_function_library_runtime.h | 110 std::vector<Device*>* output_devices) const;
243 const std::vector<string>& output_devices,
|
| /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/kernels/ |
| D | partitioned_function_ops.cc | 144 opts->output_devices.push_back(cpu_device.name()); in FillOutputDevices()
146 opts->output_devices.push_back(opts->target); in FillOutputDevices()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 531 std::vector<string> output_devices; member
|
| D | function.cc | 906 for (int i = 0; i < options.output_devices.size(); ++i) { in Canonicalize()
908 "_output_dev", i, "=", str_util::CEscape(options.output_devices[i]))); in Canonicalize()
|