Searched refs:max_intra_op_parallelism_ (Results 1 – 6 of 6) sorted by relevance
64 max_intra_op_parallelism_(max_intra_op_parallelism) {} in ThreadPoolResource()68 if (max_intra_op_parallelism_ < 0) { in Schedule()75 ScopedPerThreadMaxParallelism scope(max_intra_op_parallelism_); in Schedule()88 const int max_intra_op_parallelism_; member in tensorflow::data::experimental::ThreadPoolResource101 &max_intra_op_parallelism_)); in ThreadPoolHandleOp()132 max_intra_op_parallelism_); in Compute()148 int max_intra_op_parallelism_; member in tensorflow::data::experimental::ThreadPoolHandleOp294 max_intra_op_parallelism_(max_intra_op_parallelism), in Dataset()298 max_intra_op_parallelism_))}}) { in Dataset()340 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::__anon15242baf0111::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
147 max_intra_op_parallelism_ = in Iterator()250 RunnerWithMaxParallelism(params.runner, max_intra_op_parallelism_); in CreateParams()279 int64_t max_intra_op_parallelism_; member in tensorflow::data::RootDataset::Iterator
1347 ::int32_t max_intra_op_parallelism_; in Swap()2487 _impl_.optional_max_intra_op_parallelism_.max_intra_op_parallelism_ = 0; in clear_max_intra_op_parallelism()2493 return _impl_.optional_max_intra_op_parallelism_.max_intra_op_parallelism_; in _internal_max_intra_op_parallelism()2502 _impl_.optional_max_intra_op_parallelism_.max_intra_op_parallelism_ = value; in _internal_set_max_intra_op_parallelism()