Function SetPerThreadMaxParallelism

tensorflow/core/util/work_sharder.cc:25–28

/* ABSL_CONST_INIT */ thread_local int per_thread_max_parallism = 1000000;

void SetPerThreadMaxParallelism(int max_parallelism) {
  CHECK_LE(0, max_parallelism);
  per_thread_max_parallism = max_parallelism;
}

int GetPerThreadMaxParallelism() { return per_thread_max_parallism; }