Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.data.experimental.-autotune-algorithm.pbtxt1 path: "tensorflow.data.experimental.AutotuneAlgorithm"
3 is_instance: "<enum \'AutotuneAlgorithm\'>"
6 mtype: "<enum \'AutotuneAlgorithm\'>"
10 mtype: "<enum \'AutotuneAlgorithm\'>"
14 mtype: "<enum \'AutotuneAlgorithm\'>"
18 mtype: "<enum \'AutotuneAlgorithm\'>"
22 mtype: "<enum \'AutotuneAlgorithm\'>"
H A Dtensorflow.data.experimental.pbtxt12 name: "AutotuneAlgorithm"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.data.experimental.-autotune-algorithm.pbtxt1 path: "tensorflow.data.experimental.AutotuneAlgorithm"
3 is_instance: "<enum \'AutotuneAlgorithm\'>"
6 mtype: "<enum \'AutotuneAlgorithm\'>"
10 mtype: "<enum \'AutotuneAlgorithm\'>"
14 mtype: "<enum \'AutotuneAlgorithm\'>"
18 mtype: "<enum \'AutotuneAlgorithm\'>"
22 mtype: "<enum \'AutotuneAlgorithm\'>"
H A Dtensorflow.data.experimental.pbtxt12 name: "AutotuneAlgorithm"
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Doptions.py29 class AutotuneAlgorithm(enum.Enum): class
56 return model_pb2.AutotuneAlgorithm.DEFAULT
58 return model_pb2.AutotuneAlgorithm.HILL_CLIMB
60 return model_pb2.AutotuneAlgorithm.GRADIENT_DESCENT
62 return model_pb2.AutotuneAlgorithm.MAX_PARALLELISM
64 return model_pb2.AutotuneAlgorithm.STAGE_BASED
71 if pb == model_pb2.AutotuneAlgorithm.DEFAULT:
73 if pb == model_pb2.AutotuneAlgorithm.HILL_CLIMB:
75 if pb == model_pb2.AutotuneAlgorithm.GRADIENT_DESCENT:
77 if pb == model_pb2.AutotuneAlgorithm.MAX_PARALLELISM:
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dmodel_dataset_op.cc52 model::AutotuneAlgorithm algorithm, int64_t cpu_budget, in Dataset()
58 model::AutotuneAlgorithm algorithm, int64_t cpu_budget, in Dataset()
214 const model::AutotuneAlgorithm algorithm_;
223 model::AutotuneAlgorithm algorithm, in MakeDatasetFromOptions()
238 algorithm_ = model::AutotuneAlgorithm(algorithm); in ModelDatasetOp()
240 algorithm_ = model::AutotuneAlgorithm::HILL_CLIMB; in ModelDatasetOp()
274 model::AutotuneAlgorithm algorithm, in MakeDatasetFromOptions()
H A Dmodel_dataset_op.h40 model::AutotuneAlgorithm algorithm,
53 model::AutotuneAlgorithm algorithm_;
72 model::AutotuneAlgorithm algorithm,
H A Dfinalize_dataset_op.cc39 model::AutotuneAlgorithm* algorithm, in GetModelDatasetParams()
41 *algorithm = model::AutotuneAlgorithm::HILL_CLIMB; in GetModelDatasetParams()
66 model::AutotuneAlgorithm algorithm; in MakeDatasetHelper()
/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/
Dmodel.pb.h107 enum AutotuneAlgorithm : int { enum
117 constexpr AutotuneAlgorithm AutotuneAlgorithm_MIN = DEFAULT;
118 constexpr AutotuneAlgorithm AutotuneAlgorithm_MAX = STAGE_BASED;
121 const std::string& AutotuneAlgorithm_Name(AutotuneAlgorithm value);
124 static_assert(::std::is_same<T, AutotuneAlgorithm>::value || in AutotuneAlgorithm_Name()
127 return AutotuneAlgorithm_Name(static_cast<AutotuneAlgorithm>(enum_t_value)); in AutotuneAlgorithm_Name()
130 ::PROTOBUF_NAMESPACE_ID::ConstStringParam name, AutotuneAlgorithm* value);
830 ::tensorflow::data::model::AutotuneAlgorithm algorithm() const; in Swap()
831 void set_algorithm(::tensorflow::data::model::AutotuneAlgorithm value); in Swap()
833 ::tensorflow::data::model::AutotuneAlgorithm _internal_algorithm() const; in Swap()
[all …]
Ddataset_options.pb.h327 ::tensorflow::data::model::AutotuneAlgorithm autotune_algorithm() const; in Swap()
328 void set_autotune_algorithm(::tensorflow::data::model::AutotuneAlgorithm value); in Swap()
330 ::tensorflow::data::model::AutotuneAlgorithm _internal_autotune_algorithm() const; in Swap()
331 void _internal_set_autotune_algorithm(::tensorflow::data::model::AutotuneAlgorithm value); in Swap()
1797 inline ::tensorflow::data::model::AutotuneAlgorithm AutotuneOptions::_internal_autotune_algorithm()… in _internal_autotune_algorithm()
1799 …return static_cast< ::tensorflow::data::model::AutotuneAlgorithm >(_impl_.optional_autotune_algori… in _internal_autotune_algorithm()
1801 return static_cast< ::tensorflow::data::model::AutotuneAlgorithm >(0); in _internal_autotune_algorithm()
1803 inline ::tensorflow::data::model::AutotuneAlgorithm AutotuneOptions::autotune_algorithm() const { in autotune_algorithm()
1807 …tuneOptions::_internal_set_autotune_algorithm(::tensorflow::data::model::AutotuneAlgorithm value) { in _internal_set_autotune_algorithm()
1814 inline void AutotuneOptions::set_autotune_algorithm(::tensorflow::data::model::AutotuneAlgorithm va… in set_autotune_algorithm()
Dmodel.pb.cc231 AutotuneAlgorithm value) { in AutotuneAlgorithm_Name()
246 ::PROTOBUF_NAMESPACE_ID::ConstStringParam name, AutotuneAlgorithm* value) { in AutotuneAlgorithm_Parse()
251 *value = static_cast<AutotuneAlgorithm>(int_value); in AutotuneAlgorithm_Parse()
1423 _internal_set_algorithm(static_cast<::tensorflow::data::model::AutotuneAlgorithm>(val)); in _InternalParse()
Ddataset_options.pb.cc535 … _internal_set_autotune_algorithm(static_cast<::tensorflow::data::model::AutotuneAlgorithm>(val)); in _InternalParse()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dmodel_test.cc1107 optimization_params.set_algorithm(AutotuneAlgorithm::GRADIENT_DESCENT); in TEST()
1251 : public ::testing::TestWithParam<model::AutotuneAlgorithm> {};
1254 const model::AutotuneAlgorithm algorithm = GetParam(); in TEST_P()
2165 model_->Optimize(AutotuneAlgorithm::STAGE_BASED, 20, 1000, 50, in TEST_F()
2217 model_->Optimize(AutotuneAlgorithm::STAGE_BASED, 20, 1000, 50, in TEST_F()
2284 model_->Optimize(AutotuneAlgorithm::STAGE_BASED, 5, 1000, 50, in TEST_F()
2354 model_->Optimize(AutotuneAlgorithm::STAGE_BASED, 10, 100, 0, in TEST_F()
2359 model_->Optimize(AutotuneAlgorithm::STAGE_BASED, 10, 100000, 0, in TEST_F()
2364 model_->Optimize(AutotuneAlgorithm::STAGE_BASED, 10, 100, 0, in TEST_F()
2422 model_->Optimize(AutotuneAlgorithm::STAGE_BASED, 20, 10000, 50, in TEST_F()
H A Dmodel.proto20 enum AutotuneAlgorithm { enum
122 AutotuneAlgorithm algorithm = 1;
H A Dmodel.cc2168 void Model::Optimize(AutotuneAlgorithm algorithm, int64_t cpu_budget, in Optimize()
2185 case AutotuneAlgorithm::DEFAULT: in Optimize()
2186 case AutotuneAlgorithm::MAX_PARALLELISM: in Optimize()
2190 case AutotuneAlgorithm::HILL_CLIMB: in Optimize()
2193 case AutotuneAlgorithm::GRADIENT_DESCENT: in Optimize()
2197 case AutotuneAlgorithm::STAGE_BASED: in Optimize()
2277 Status Model::OptimizeLoop(AutotuneAlgorithm algorithm, int64_t cpu_budget, in OptimizeLoop()
2313 if (algorithm == AutotuneAlgorithm::STAGE_BASED) { in OptimizeLoop()
H A Dmodel.h803 Status OptimizeLoop(AutotuneAlgorithm algorithm, int64_t cpu_budget,
809 void Optimize(AutotuneAlgorithm algorithm, int64_t cpu_budget,
H A Ddataset_options.proto55 model.AutotuneAlgorithm autotune_algorithm = 4;
/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()
H A Droot_dataset.h32 model::AutotuneAlgorithm autotune_algorithm;
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/
H A D__init__.py151 from tensorflow.python.data.ops.options import AutotuneAlgorithm
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_java_lite/android_common_apex33/turbine/
Dtensorflow_core_proto_java_lite.jarorg/tensorflow/distruntime/ClusterDef$Builder.class <Unknown> package org.tensorflow. ...
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_java_lite/android_common_apex33/javac/
Dtensorflow_core_proto_java_lite.jarMETA-INF/ META-INF/MANIFEST.MF com/ com/google/ com/ ...
/aosp_15_r20/out/soong/.intermediates/packages/modules/OnDevicePersonalization/federatedcompute/apk/FederatedCompute/android_common_apex33/combined/
DFederatedCompute.jarMETA-INF/ META-INF/MANIFEST.MF META-INF/androidx/ META- ...
/aosp_15_r20/out/soong/.intermediates/packages/modules/OnDevicePersonalization/federatedcompute/apk/FederatedCompute/android_common_apex33/jarjar/
DFederatedCompute.jarxla/Protocol.class Protocol.java package xla public final xla.Protocol extends java.lang ...
/aosp_15_r20/out/soong/.intermediates/packages/modules/OnDevicePersonalization/OnDevicePersonalization/android_common_apex33/withres/
DOnDevicePersonalization.jarMETA-INF/ META-INF/MANIFEST.MF META-INF/androidx.annotation_annotation- ...