Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/api/golden/v2/
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\'>"
Dtensorflow.data.experimental.pbtxt12 name: "AutotuneAlgorithm"
/external/tensorflow/tensorflow/tools/api/golden/v1/
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\'>"
Dtensorflow.data.experimental.pbtxt12 name: "AutotuneAlgorithm"
/external/tensorflow/tensorflow/python/data/ops/
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 …]
/external/tensorflow/tensorflow/core/kernels/data/
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()
Dmodel_dataset_op.h40 model::AutotuneAlgorithm algorithm,
53 model::AutotuneAlgorithm algorithm_;
72 model::AutotuneAlgorithm algorithm,
Dfinalize_dataset_op.cc39 model::AutotuneAlgorithm* algorithm, in GetModelDatasetParams()
41 *algorithm = model::AutotuneAlgorithm::HILL_CLIMB; in GetModelDatasetParams()
66 model::AutotuneAlgorithm algorithm; in MakeDatasetHelper()
/external/tensorflow/tensorflow/core/framework/
Dmodel.proto20 enum AutotuneAlgorithm { enum
122 AutotuneAlgorithm algorithm = 1;
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()
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()
Dmodel.h803 Status OptimizeLoop(AutotuneAlgorithm algorithm, int64_t cpu_budget,
809 void Optimize(AutotuneAlgorithm algorithm, int64_t cpu_budget,
Ddataset_options.proto55 model.AutotuneAlgorithm autotune_algorithm = 4;
/external/tensorflow/tensorflow/core/data/
Droot_dataset.h32 model::AutotuneAlgorithm autotune_algorithm;
Droot_dataset.cc65 params->autotune_algorithm = model::AutotuneAlgorithm::DEFAULT; in SetRootDatasetParams()
67 params->autotune_algorithm = model::AutotuneAlgorithm::STAGE_BASED; in SetRootDatasetParams()
/external/tensorflow/tensorflow/python/data/experimental/
D__init__.py151 from tensorflow.python.data.ops.options import AutotuneAlgorithm