Searched refs:kUseInterOpParallelism (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_dataset_op.cc | 38 /* static */ constexpr const char* const MapDatasetOp::kUseInterOpParallelism; member in tensorflow::data::MapDatasetOp
127 std::make_pair(kUseInterOpParallelism, use_inter_op_parallelism_attr), in AsGraphDefInternal()
217 OP_REQUIRES_OK(ctx, ctx->GetAttr(kUseInterOpParallelism, in MapDatasetOp()
|
| D | map_dataset_op.h | 33 static constexpr const char* const kUseInterOpParallelism = variable
|
| D | parallel_map_dataset_op.h | 35 static constexpr const char* const kUseInterOpParallelism = variable
|
| D | parallel_map_dataset_op.cc | 52 ParallelMapDatasetOp::kUseInterOpParallelism; member in tensorflow::data::ParallelMapDatasetOp
172 attrs.emplace_back(kUseInterOpParallelism, use_inter_op_parallelism_attr); in AsGraphDefInternal()
678 OP_REQUIRES_OK(ctx, ctx->GetAttr(kUseInterOpParallelism, in ParallelMapDatasetOp()
|
| D | parallel_map_dataset_op_test.cc | 77 {ParallelMapDatasetOp::kUseInterOpParallelism, in GetAttributes()
|
| D | dataset_test_base.cc | 1054 {MapDatasetOp::kUseInterOpParallelism, use_inter_op_parallelism_}, in GetAttributes()
|