Home
last modified time | relevance | path

Searched refs:autotune_algorithm (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Droot_dataset.cc65 params->autotune_algorithm = model::AutotuneAlgorithm::DEFAULT; in SetRootDatasetParams()
67 params->autotune_algorithm = model::AutotuneAlgorithm::STAGE_BASED; in SetRootDatasetParams()
71 params->autotune_algorithm = in SetRootDatasetParams()
72 options.autotune_options().autotune_algorithm(); in SetRootDatasetParams()
86 kAlgorithm, model::AutotuneAlgorithm_Name(params.autotune_algorithm))); in AddTraceMetadata()
261 model_->OptimizeLoop(dataset()->params_.autotune_algorithm, in EnsureModelThreadStarted()
H A Droot_dataset.h32 model::AutotuneAlgorithm autotune_algorithm; member
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Doptions.py224 autotune_algorithm = options_lib.create_option( variable in AutotuneOptions
238 if self.autotune_algorithm is not None:
239 pb.autotune_algorithm = AutotuneAlgorithm._to_proto( # pylint: disable=protected-access
240 self.autotune_algorithm)
251 self.autotune_algorithm = AutotuneAlgorithm._from_proto( # pylint: disable=protected-access
252 pb.autotune_algorithm)
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.data.experimental.-autotune-options.pbtxt7 name: "autotune_algorithm"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.data.experimental.-autotune-options.pbtxt7 name: "autotune_algorithm"
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddataset_options.proto55 model.AutotuneAlgorithm autotune_algorithm = 4; field
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Ddataset_options.pb.h327 ::tensorflow::data::model::AutotuneAlgorithm autotune_algorithm() const; in Swap()
1803 inline ::tensorflow::data::model::AutotuneAlgorithm AutotuneOptions::autotune_algorithm() const { in autotune_algorithm() function
/aosp_15_r20/external/tensorflow/
H A DRELEASE.md86 …* Added a new value to `tf.data.Options.autotune.autotune_algorithm`: STAGE_BASED. If the autotu…
5815 * Removed `autotune_algorithm` from experimental optimization options.