Searched refs:run_opts (Results 1 – 4 of 4) sorted by relevance
214 FunctionLibraryRuntime::Options run_opts; in RunFunction() local215 run_opts.step_id = ctx->step_id(); in RunFunction()216 run_opts.step_container = ctx->step_container(); in RunFunction()217 run_opts.cancellation_manager = ctx->cancellation_manager(); in RunFunction()218 run_opts.stats_collector = ctx->stats_collector(); in RunFunction()219 run_opts.collective_executor = ctx->collective_executor(); in RunFunction()222 run_opts.runner = ctx->runner(); in RunFunction()223 run_opts.source_device = in RunFunction()225 run_opts.allow_dead_tensors = true; in RunFunction()229 run_opts.rendezvous = rendez; in RunFunction()[all …]
417 void ExecutorArgsFromOptions(const FunctionLibraryRuntime::Options& run_opts,932 const FunctionLibraryRuntime::Options& run_opts, CallFrameInterface* frame, in ExecutorArgsFromOptions() argument935 exec_args->step_id = run_opts.step_id; in ExecutorArgsFromOptions()936 exec_args->rendezvous = run_opts.rendezvous; in ExecutorArgsFromOptions()937 exec_args->stats_collector = run_opts.stats_collector; in ExecutorArgsFromOptions()938 exec_args->cancellation_manager = run_opts.cancellation_manager; in ExecutorArgsFromOptions()939 exec_args->step_container = run_opts.step_container; in ExecutorArgsFromOptions()940 if (run_opts.runner) { in ExecutorArgsFromOptions()941 exec_args->runner = *run_opts.runner; in ExecutorArgsFromOptions()945 exec_args->collective_executor = run_opts.collective_executor; in ExecutorArgsFromOptions()[all …]
1038 RunOptions run_opts; in TEST() local1039 run_opts.set_inter_op_thread_pool(-1); in TEST()1040 auto s = sess->Run(run_opts, {}, {y->name() + ":0"}, {}, &outputs, nullptr); in TEST()
303 ExecutableRunOptions run_opts; in BuildAndRunComputation() local304 run_opts.set_allocator(client_->backend().memory_allocator()); in BuildAndRunComputation()305 run_opts.set_intra_op_thread_pool( in BuildAndRunComputation()308 executable->Run({&input_data}, run_opts)); in BuildAndRunComputation()