Searched defs:use_inter_op_parallelism (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
H A D | captured_function.h | 75 bool use_inter_op_parallelism = true; member 109 bool use_inter_op_parallelism() const { return use_inter_op_parallelism_; } in use_inter_op_parallelism() function 209 bool use_inter_op_parallelism() const { in use_inter_op_parallelism() function
|
H A D | dataset_test_base.h | 266 bool use_inter_op_parallelism, bool preserve_cardinality, in MapDatasetParams()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/data/benchmarks/ |
H A D | map_benchmark.py | 33 def benchmark_helper(chain_length, fn, use_inter_op_parallelism, label, argument 72 def benchmark_helper(fan_out, fn, use_inter_op_parallelism, label, argument
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
H A D | reduce_dataset_op_test.cc | 34 bool use_inter_op_parallelism, string node_name) in ReduceDatasetParams()
|
H A D | parallel_map_dataset_op_test.cc | 33 bool use_inter_op_parallelism, const std::string& deterministic, in ParallelMapDatasetParams()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
H A D | prefetching_ops.py | 234 def __init__(self, input_dataset, map_func, use_inter_op_parallelism=True): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/ |
H A D | dataset_ops.py | 5398 use_inter_op_parallelism=True, argument 5440 use_inter_op_parallelism=True, argument
|