Searched refs:use_inter_op_parallelism_ (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
D | parallel_map_dataset_op.cc | 45 &use_inter_op_parallelism_)); in ParallelMapDatasetOp() 66 use_inter_op_parallelism_, in MakeDataset() 74 output_shapes_, use_inter_op_parallelism_, sloppy_, in MakeDataset() 95 use_inter_op_parallelism_(use_inter_op_parallelism), in Dataset() 178 b->BuildAttrValue(use_inter_op_parallelism_, in AsGraphDefInternal() 258 if (!dataset_->use_inter_op_parallelism_) { in MapFunc() 278 const bool use_inter_op_parallelism_; member in tensorflow::data::__anon58e4a4830111::ParallelMapDatasetOp::Dataset 288 bool use_inter_op_parallelism_; member in tensorflow::data::__anon58e4a4830111::ParallelMapDatasetOp
|
D | map_dataset_op.cc | 41 &use_inter_op_parallelism_)); in MapDatasetOp() 51 use_inter_op_parallelism_, in MakeDataset() 91 output_shapes_, use_inter_op_parallelism_, in MakeDataset() 108 use_inter_op_parallelism_(use_inter_op_parallelism), in Dataset() 171 b->BuildAttrValue(use_inter_op_parallelism_, in AsGraphDefInternal() 265 const bool use_inter_op_parallelism_; member in tensorflow::data::__anonb3c9acd70111::MapDatasetOp::Dataset 276 bool use_inter_op_parallelism_; member in tensorflow::data::__anonb3c9acd70111::MapDatasetOp
|
D | captured_function.h | 161 const bool use_inter_op_parallelism_; variable
|
D | captured_function.cc | 139 if (!use_inter_op_parallelism_) { in Instantiate() 489 use_inter_op_parallelism_(use_inter_op_parallelism) {} in CapturedFunction()
|
D | iterator_ops.cc | 647 &use_inter_op_parallelism_)); in ReduceDatasetOp() 667 use_inter_op_parallelism_, &captured_func), in ComputeAsync() 752 bool use_inter_op_parallelism_; member in tensorflow::data::__anon9c98ff710411::ReduceDatasetOp
|