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 | 431 const FunctionLibraryRuntime::InstantiateOptions& inst_opts, in TestTwoDeviceMult() argument
439 inst_opts, {x}, {&y_cpu, &y_gpu}); in TestTwoDeviceMult()
462 const FunctionLibraryRuntime::InstantiateOptions& inst_opts) { in TestTwoDeviceInputOutput() argument
469 if (str_util::StrContains(inst_opts.input_devices[0], "GPU")) { in TestTwoDeviceInputOutput()
473 if (str_util::StrContains(inst_opts.input_devices[1], "GPU")) { in TestTwoDeviceInputOutput()
480 inst_opts, {x1, x2}, {&y1, &y2})); in TestTwoDeviceInputOutput()
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()
513 FunctionLibraryRuntime::InstantiateOptions inst_opts; in MakeOptions() local
514 inst_opts.target = target; in MakeOptions()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 136 FunctionLibraryRuntime::InstantiateOptions inst_opts; in Instantiate() local
137 inst_opts.overlay_lib = ctx->function_library().get(); in Instantiate()
138 inst_opts.create_kernels_eagerly = true; in Instantiate()
140 inst_opts.executor_type = "SINGLE_THREADED_EXECUTOR"; in Instantiate()
145 func_.name(), AttrSlice(&func_.attr()), inst_opts, &f_handle)); in Instantiate()
|