MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / SetRunOptions

Method SetRunOptions

tensorflow/core/kernels/data/map_defun_op.cc:240–253  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

238}
239
240void MapDefunOp::SetRunOptions(OpKernelContext* ctx,
241 FunctionLibraryRuntime::Options* opts,
242 ComputeOptions* compute_opts,
243 bool always_collect_stats) {
244 opts->rendezvous = ctx->rendezvous();
245 if (always_collect_stats) {
246 opts->stats_collector = ctx->stats_collector();
247 }
248 if (max_intra_op_parallelism_ >= 1) {
249 opts->runner = &compute_opts->runner;
250 } else {
251 opts->runner = ctx->runner();
252 }
253}
254
255Status MapDefunOp::SetupArgs(OpKernelContext* ctx,
256 ComputeOptions** compute_opts) {

Callers

nothing calls this directly

Calls 3

rendezvousMethod · 0.80
stats_collectorMethod · 0.80
runnerMethod · 0.45

Tested by

no test coverage detected