Searched refs:max_intra_op_parallelism_ (Results 1 – 5 of 5) sorted by relevance
48 max_intra_op_parallelism_(max_intra_op_parallelism) {} in ThreadPoolResource()52 if (max_intra_op_parallelism_ < 0) { in Schedule()59 ScopedPerThreadMaxParallelism scope(max_intra_op_parallelism_); in Schedule()72 const int max_intra_op_parallelism_; member in tensorflow::data::experimental::ThreadPoolResource85 &max_intra_op_parallelism_)); in ThreadPoolHandleOp()118 max_intra_op_parallelism_); in Compute()134 int max_intra_op_parallelism_; member in tensorflow::data::experimental::ThreadPoolHandleOp278 max_intra_op_parallelism_(max_intra_op_parallelism), in Dataset()282 max_intra_op_parallelism_))}}) { in Dataset()324 TF_RETURN_IF_ERROR(b->AddScalar(max_intra_op_parallelism_, in AsGraphDefInternal()[all …]
41 max_intra_op_parallelism_(max_intra_op_parallelism) {} in MapDefunOpParams()72 {MapDefunOp::kMaxIntraOpParallelism, max_intra_op_parallelism_}}; in GetAttributes()87 int max_intra_op_parallelism_; member in tensorflow::data::__anonda3d45650111::MapDefunOpParams
180 ctx, ctx->GetAttr(kMaxIntraOpParallelism, &max_intra_op_parallelism_)); in MapDefunOp()249 if (max_intra_op_parallelism_ >= 1) { in SetRunOptions()289 batch_size, output_shapes_, max_intra_op_parallelism_); in SetupArgs()
70 int max_intra_op_parallelism_; variable
80 max_intra_op_parallelism_ = in Iterator()148 RunnerWithMaxParallelism(params.runner, max_intra_op_parallelism_); in CreateParams()176 int64 max_intra_op_parallelism_; member in tensorflow::data::RootDataset::Iterator