Home
last modified time | relevance | path

Searched defs:use_inter_op_parallelism (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dcaptured_function.h75 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 Ddataset_test_base.h266 bool use_inter_op_parallelism, bool preserve_cardinality, in MapDatasetParams()
/aosp_15_r20/external/tensorflow/tensorflow/python/data/benchmarks/
H A Dmap_benchmark.py33 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 Dreduce_dataset_op_test.cc34 bool use_inter_op_parallelism, string node_name) in ReduceDatasetParams()
H A Dparallel_map_dataset_op_test.cc33 bool use_inter_op_parallelism, const std::string& deterministic, in ParallelMapDatasetParams()
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Dprefetching_ops.py234 def __init__(self, input_dataset, map_func, use_inter_op_parallelism=True): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Ddataset_ops.py5398 use_inter_op_parallelism=True, argument
5440 use_inter_op_parallelism=True, argument