Searched refs:inst_opts (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime_test.cc | 562 const FunctionLibraryRuntime::InstantiateOptions& inst_opts, in TestTwoDeviceMult() argument
570 inst_opts, {x}, {&y_cpu, &y_gpu}); in TestTwoDeviceMult()
591 const FunctionLibraryRuntime::InstantiateOptions& inst_opts) { in TestTwoDeviceInputOutput() argument
599 if (absl::StrContains(inst_opts.input_devices[0], "GPU")) { in TestTwoDeviceInputOutput()
603 if (absl::StrContains(inst_opts.input_devices[1], "GPU")) { in TestTwoDeviceInputOutput()
609 inst_opts, {x1, x2}, {&y1, &y2})); in TestTwoDeviceInputOutput()
611 if (absl::StrContains(inst_opts.output_devices[0], "GPU")) { in TestTwoDeviceInputOutput()
619 if (absl::StrContains(inst_opts.output_devices[1], "GPU")) { in TestTwoDeviceInputOutput()
640 FunctionLibraryRuntime::InstantiateOptions inst_opts; in MakeOptions() local
641 inst_opts.target = target; in MakeOptions()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 593 FunctionLibraryRuntime::InstantiateOptions inst_opts; in Instantiate() local
594 inst_opts.lib_def = metadata_->lib_def(); in Instantiate()
595 inst_opts.create_kernels_eagerly = true; in Instantiate()
596 inst_opts.default_device_to_target = metadata_->use_default_device(); in Instantiate()
597 inst_opts.config_proto = in Instantiate()
600 inst_opts.executor_type = "SINGLE_THREADED_EXECUTOR"; in Instantiate()
602 inst_opts.is_multi_device_function = metadata_->use_multi_device_function(); in Instantiate()
606 inst_opts.target = lib->device()->name(); in Instantiate()
611 if (inst_opts.is_multi_device_function) { in Instantiate()
623 inst_opts.input_devices.push_back(inst_opts.target); in Instantiate()
[all …]
|