Home
last modified time | relevance | path

Searched refs:kUseInterOpParallelism (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dmap_dataset_op.cc39 /* static */ constexpr const char* const MapDatasetOp::kUseInterOpParallelism; member in tensorflow::data::MapDatasetOp
149 std::make_pair(kUseInterOpParallelism, use_inter_op_parallelism_attr), in AsGraphDefInternal()
242 OP_REQUIRES_OK(ctx, ctx->GetAttr(kUseInterOpParallelism, in MapDatasetOp()
H A Dmap_dataset_op.h33 static constexpr const char* const kUseInterOpParallelism = variable
H A Dparallel_map_dataset_op.cc52 ParallelMapDatasetOp::kUseInterOpParallelism; member in tensorflow::data::ParallelMapDatasetOp
206 attrs.emplace_back(kUseInterOpParallelism, use_inter_op_parallelism_attr); in AsGraphDefInternal()
726 OP_REQUIRES_OK(ctx, ctx->GetAttr(kUseInterOpParallelism, in ParallelMapDatasetOp()
H A Dparallel_map_dataset_op.h35 static constexpr const char* const kUseInterOpParallelism = variable