Home
last modified time | relevance | path

Searched refs:run_opts (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc237 FunctionLibraryRuntime::Options run_opts; in RunFunction() local
240 run_opts.step_id, [resource_mgr](const string& name) { in RunFunction()
243 run_opts.step_container = step_container; in RunFunction()
244 run_opts.cancellation_manager = ctx->cancellation_manager(); in RunFunction()
245 run_opts.stats_collector = ctx->stats_collector(); in RunFunction()
246 run_opts.collective_executor = ctx->collective_executor(); in RunFunction()
249 run_opts.runner = ctx->runner(); in RunFunction()
250 run_opts.run_all_kernels_inline = ctx->run_all_kernels_inline(); in RunFunction()
251 run_opts.source_device = in RunFunction()
253 run_opts.allow_dead_tensors = true; in RunFunction()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.cc448 Handle handle, Options* run_opts, Item** out_item,
451 void ExecutorArgsFromOptions(const FunctionLibraryRuntime::Options& run_opts,
979 const FunctionLibraryRuntime::Options& run_opts, CallFrameInterface* frame, in ExecutorArgsFromOptions() argument
982 exec_args->step_id = run_opts.step_id; in ExecutorArgsFromOptions()
983 exec_args->rendezvous = run_opts.rendezvous; in ExecutorArgsFromOptions()
984 exec_args->stats_collector = run_opts.stats_collector; in ExecutorArgsFromOptions()
985 exec_args->cancellation_manager = run_opts.cancellation_manager; in ExecutorArgsFromOptions()
986 exec_args->step_container = run_opts.step_container; in ExecutorArgsFromOptions()
987 if (run_opts.runner) { in ExecutorArgsFromOptions()
988 exec_args->runner = *run_opts.runner; in ExecutorArgsFromOptions()
[all …]
Ddirect_session_test.cc1206 RunOptions run_opts; in TEST() local
1207 run_opts.set_inter_op_thread_pool(-1); in TEST()
1208 auto s = sess->Run(run_opts, {}, {y->name() + ":0"}, {}, &outputs, nullptr); in TEST()
1228 RunOptions run_opts; in TEST() local
1229 run_opts.set_inter_op_thread_pool(-1); in TEST()
1230 auto s = sess->Run(run_opts, {}, {y->name() + ":0"}, {}, &outputs, nullptr); in TEST()
1251 RunOptions run_opts; in TEST() local
1252 run_opts.set_inter_op_thread_pool(-1); in TEST()
1253 auto s = sess->Run(run_opts, {}, {y->name() + ":0"}, {}, &outputs, nullptr); in TEST()
1282 RunOptions run_opts; in TEST() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dexhaustive_op_test_utils.h269 ExecutableRunOptions run_opts;
270 run_opts.set_allocator(client_->backend().memory_allocator());
271 run_opts.set_intra_op_thread_pool(
274 executables[0]->Run(input_buffer_pointers, run_opts));
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddriver.cc58 extern void EntryModule(char* result_buffer, char* run_opts, char** params,
/external/tensorflow/tensorflow/python/kernel_tests/
Dcontrol_flow_ops_py_test.py1071 run_opts = config_pb2.RunOptions(
1084 r, options=run_opts, run_metadata=run_metadata_with_lowering)
1091 r, options=run_opts, run_metadata=run_metadata_no_lowering)