Home
last modified time | relevance | path

Searched refs:use_inter_op_parallelism (Results 1 – 25 of 42) sorted by relevance

12

/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
38 dataset, fn, use_inter_op_parallelism=use_inter_op_parallelism)
53 use_inter_op_parallelism=True,
59 use_inter_op_parallelism=False,
65 use_inter_op_parallelism=True,
72 def benchmark_helper(fan_out, fn, use_inter_op_parallelism, label, argument
77 dataset, fn, use_inter_op_parallelism=use_inter_op_parallelism)
91 use_inter_op_parallelism=True,
97 use_inter_op_parallelism=False,
103 use_inter_op_parallelism=True,
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DParallelMapDataset.pbtxt122 name: "use_inter_op_parallelism"
169 name: "use_inter_op_parallelism"
223 name: "use_inter_op_parallelism"
284 name: "use_inter_op_parallelism"
359 name: "use_inter_op_parallelism"
447 name: "use_inter_op_parallelism"
H A DMapDataset.pbtxt110 name: "use_inter_op_parallelism"
153 name: "use_inter_op_parallelism"
203 name: "use_inter_op_parallelism"
267 name: "use_inter_op_parallelism"
344 name: "use_inter_op_parallelism"
H A DExperimentalMapDataset.pbtxt37 name: "use_inter_op_parallelism"
80 name: "use_inter_op_parallelism"
137 name: "use_inter_op_parallelism"
207 name: "use_inter_op_parallelism"
H A DParallelMapDatasetV2.pbtxt41 name: "use_inter_op_parallelism"
102 name: "use_inter_op_parallelism"
177 name: "use_inter_op_parallelism"
265 name: "use_inter_op_parallelism"
H A DReduceDataset.pbtxt47 name: "use_inter_op_parallelism"
100 name: "use_inter_op_parallelism"
154 name: "use_inter_op_parallelism"
/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
118 use_inter_op_parallelism_(params.use_inter_op_parallelism) {} in FunctionMetadata()
209 bool use_inter_op_parallelism() const { in use_inter_op_parallelism() function
210 return metadata_->use_inter_op_parallelism(); in use_inter_op_parallelism()
H A Dstandalone_test.cc213 key: "use_inter_op_parallelism"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
H A DParallelMapDataset.pbtxt122 name: "use_inter_op_parallelism"
169 name: "use_inter_op_parallelism"
223 name: "use_inter_op_parallelism"
H A DExperimentalMapDataset.pbtxt37 name: "use_inter_op_parallelism"
80 name: "use_inter_op_parallelism"
H A DReduceDataset.pbtxt47 name: "use_inter_op_parallelism"
100 name: "use_inter_op_parallelism"
H A DMapDataset.pbtxt110 name: "use_inter_op_parallelism"
153 name: "use_inter_op_parallelism"
H A DParallelMapDatasetV2.pbtxt41 name: "use_inter_op_parallelism"
/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
237 self._use_inter_op_parallelism = use_inter_op_parallelism
248 use_inter_op_parallelism=self._use_inter_op_parallelism,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
H A Ddecompose_reduce_dataset.mlir14 …output_types = [i64], use_inter_op_parallelism = true } : (tensor<!tf_type.variant>, tensor<i64>) …
53 …output_types = [i64], use_inter_op_parallelism = true, _xla_compile_device_type="TPU"} : (tensor<!…
93 …output_types = [i64], use_inter_op_parallelism = true, _xla_compile_device_type="TPU"} : (tensor<!…
136 …output_types = [i64, i32], use_inter_op_parallelism = true, _xla_compile_device_type="TPU"} : (ten…
H A Dtf_data_fuse_pmap_and_batch.mlir18 use_inter_op_parallelism = true,
H A Dtf_data_fuse_map_and_batch.mlir18 use_inter_op_parallelism = true,
/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() argument
43 use_inter_op_parallelism_(use_inter_op_parallelism) { in ReduceDatasetParams()
H A Dmap_dataset_op.cc137 b->BuildAttrValue(captured_func_->use_inter_op_parallelism(), in AsGraphDefInternal()
243 &params.use_inter_op_parallelism)); in MapDatasetOp()
H A Dparallel_map_dataset_op_test.cc33 bool use_inter_op_parallelism, const std::string& deterministic, in ParallelMapDatasetParams() argument
42 use_inter_op_parallelism_(use_inter_op_parallelism), in ParallelMapDatasetParams()
H A Dparallel_map_dataset_op.cc204 b->BuildAttrValue(captured_func_->use_inter_op_parallelism(), in AsGraphDefInternal()
482 if (dataset()->captured_func_->use_inter_op_parallelism()) { in CallFunction()
727 &params.use_inter_op_parallelism)); in ParallelMapDatasetOp()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/
H A Dfunc_attributes.mlir20 …rt.shape<>], output_types = [i64], preserve_cardinality = true, use_inter_op_parallelism = true} {…
21 …pe.shape<>], output_types = [i64], preserve_cardinality = true, use_inter_op_parallelism = true, m…
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
H A Dmodel_dataset_test.py52 use_inter_op_parallelism=False)
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/
H A Dtest25.pbtxt741 key: "use_inter_op_parallelism"
856 key: "use_inter_op_parallelism"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtf_data_optimization.td27 $output_shapes, $use_inter_op_parallelism, $preserve_cardinality,

12