| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node_test.cc | 34 TestKernelAndDeviceFunc(std::vector<Device*> input_devices, in TestKernelAndDeviceFunc() argument
42 test_input_devices_(std::move(input_devices)) {}
106 std::vector<Device*> input_devices; in TEST() local
108 input_devices.push_back(h->DeviceOrHostCPU(*ctx)); in TEST()
111 new TestKernelAndDeviceFunc(std::move(input_devices), device0)); in TEST()
|
| D | kernel_and_device.h | 262 std::vector<Device*> input_devices, in KernelAndDeviceFunc() argument
277 input_devices_(std::move(input_devices)), in KernelAndDeviceFunc()
|
| D | kernel_and_device.cc | 161 options.input_devices.push_back(device->name()); in InstantiateFunc()
|
| /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/compiler/mlir/tensorflow/transforms/ |
| D | cluster_tf_ops_pass.cc | 97 llvm::SmallVector<std::string, 4> input_devices; member
139 func_metadata.input_devices.push_back(value_device); in GetFunctionMetadatas()
211 for (int i : llvm::seq<int>(0, metadata.input_devices.size())) { in CreateFunctions()
213 StringAttr::get(context, metadata.input_devices[i])); in CreateFunctions()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | inline_function_utils.cc | 123 std::vector<string> input_devices(caller.in_edges().size()); in InputDevices() local
131 input_devices[edge->dst_input()] = input_device; in InputDevices()
138 for (int i = 0; i < input_devices.size(); ++i) { in InputDevices()
141 << " device: " << input_devices[i] in InputDevices()
146 return input_devices; in InputDevices()
|
| D | process_function_library_runtime_test.cc | 599 if (absl::StrContains(inst_opts.input_devices[0], "GPU")) { in TestTwoDeviceInputOutput()
603 if (absl::StrContains(inst_opts.input_devices[1], "GPU")) { in TestTwoDeviceInputOutput()
638 const string& target, const std::vector<string>& input_devices, in MakeOptions() argument
642 inst_opts.input_devices = CompleteDevices(input_devices); in MakeOptions()
|
| D | process_function_library_runtime.cc | 423 const std::vector<string>& input_devices, in PinArgsAndRets() argument
436 node->set_assigned_device_name(input_devices[index]); in PinArgsAndRets()
612 if (options.input_devices.size() != signature.input_arg_size()) { in ValidateMultiDeviceOptions()
616 options.input_devices.size(), in ValidateMultiDeviceOptions()
690 for (const string& device : options.input_devices) { in InstantiateMultiDevice()
746 options.input_devices, options.output_devices, *dev_set, arg_nodes, in InstantiateMultiDevice()
|
| D | process_function_library_runtime.h | 384 Status PinArgsAndRets(const std::vector<string>& input_devices,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 353 opts.input_devices.push_back(cpu_device->name()); in InstantiateFunction()
361 opts.input_devices.push_back(rhandle.device()); in InstantiateFunction()
363 opts.input_devices.push_back(cpu_device->name()); in InstantiateFunction()
371 if (opts.input_devices.size() != signature.input_arg_size()) { in InstantiateFunction()
374 opts.input_devices.size(), " argument(s) were passed"); in InstantiateFunction()
|
| D | partitioned_function_ops.cc | 219 opts.input_devices.push_back(handle.device()); in Instantiate()
221 opts.input_devices.push_back(opts.target); in Instantiate()
|
| /external/adhd/cras/client/cras_tests/src/ |
| D | control.rs | 94 for dev in cras_client.input_devices() { in control()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 623 inst_opts.input_devices.push_back(inst_opts.target); in Instantiate()
654 inst_opts.input_devices.push_back(composite_device); in Instantiate()
656 inst_opts.input_devices.push_back(handle0.device()); in Instantiate()
666 inst_opts.input_devices.push_back(cpu_device->name()); in Instantiate()
669 inst_opts.input_devices.push_back(inst_opts.target); in Instantiate()
|
| /external/adhd/cras/client/libcras/src/ |
| D | libcras.rs | 336 pub fn input_devices(&self) -> impl Iterator<Item = CrasIodevInfo> { in input_devices() method
337 self.server_state.input_devices() in input_devices()
|
| D | cras_shm.rs | 689 pub fn input_devices(&self) -> impl Iterator<Item = CrasIodevInfo> { in input_devices() method
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.cc | 1025 options.input_devices.size()); in Canonicalize()
1035 for (int i = 0; i < options.input_devices.size(); ++i) { in Canonicalize()
1036 entries.push_back(AttrKeyAndValue("_input_dev", i, options.input_devices[i], in Canonicalize()
|
| D | function.h | 637 std::vector<string> input_devices; member
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 307 inst_ops.input_devices.resize(num_inputs, device_name); in ConstructFunctionHandle()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 842 options.input_devices.push_back(local_device_); in TEST_F()
|