Home
last modified time | relevance | path

Searched refs:max_intra_op_parallelism_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dthreadpool_dataset_op.cc48 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::ThreadPoolResource
85 &max_intra_op_parallelism_)); in ThreadPoolHandleOp()
118 max_intra_op_parallelism_); in Compute()
134 int max_intra_op_parallelism_; member in tensorflow::data::experimental::ThreadPoolHandleOp
278 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 …]
/external/tensorflow/tensorflow/core/kernels/data/
Dmap_defun_op_test.cc41 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
Dmap_defun_op.cc180 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()
Dmap_defun_op.h70 int max_intra_op_parallelism_; variable
/external/tensorflow/tensorflow/core/data/
Droot_dataset.cc80 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