Home
last modified time | relevance | path

Searched defs:cpu_budget (Results 1 – 8 of 8) sorted by relevance

/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()
224 int64_t cpu_budget, in MakeDatasetFromOptions()
275 bool cpu_budget, bool ram_budget, in MakeDatasetFromOptions()
H A Dfinalize_dataset_op.cc40 int64_t* cpu_budget, int64_t* ram_budget) { in GetModelDatasetParams()
67 int64_t cpu_budget; in MakeDatasetHelper() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dmodel.proto125 int64 cpu_budget = 2; field
H A Ddataset_options.proto40 int32 cpu_budget = 2; field
H A Dmodel.cc2168 void Model::Optimize(AutotuneAlgorithm algorithm, int64_t cpu_budget, in Optimize()
2252 bool Model::ShouldStop(int64_t cpu_budget, int64_t ram_budget, in ShouldStop()
2277 Status Model::OptimizeLoop(AutotuneAlgorithm algorithm, int64_t cpu_budget, in OptimizeLoop()
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Doptions.py208 cpu_budget = options_lib.create_option( variable in AutotuneOptions
/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.h1650 inline ::int64_t ModelProto_OptimizationParams::cpu_budget() const { in cpu_budget() function
Ddataset_options.pb.h1734 inline ::int32_t AutotuneOptions::cpu_budget() const { in cpu_budget() function