Searched refs:output_devices (Results 1 – 8 of 8) 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 | 365 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 …]
|
D | process_function_library_runtime.h | 147 std::vector<Device*>* output_devices) const; 375 const std::vector<string>& output_devices,
|
D | process_function_library_runtime_test.cc | 554 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/ |
D | partitioned_function_ops.cc | 153 opts->output_devices.push_back(""); in FillOutputDevices() 155 opts->output_devices.push_back(opts->target); in FillOutputDevices()
|
/external/tensorflow/tensorflow/core/framework/ |
D | function.h | 573 std::vector<string> output_devices; member
|
D | function.cc | 1014 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/ |
D | captured_function.cc | 591 inst_opts.output_devices.push_back(inst_opts.target); in Instantiate()
|