Home
last modified time | relevance | path

Searched refs:autotune_ (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Denable_gradient_descent.h43 autotune_ = true; in Init()
45 autotune_ = false; in Init()
61 bool autotune_ = true;
Dbatch_parallelization.h43 autotune_ = true; in Init()
45 autotune_ = false; in Init()
61 bool autotune_ = true;
Ddisable_prefetch_legacy_autotune.h43 autotune_ = true; in Init()
45 autotune_ = false; in Init()
61 bool autotune_ = true;
Dautotune_buffer_sizes.h51 autotune_ = true; in Init()
53 autotune_ = false; in Init()
69 bool autotune_ = true;
Dmap_parallelization.h43 autotune_ = true; in Init()
45 autotune_ = false; in Init()
61 bool autotune_ = true;
Ddisable_prefetch_legacy_autotune.cc41 if (!autotune_) { in OptimizeAndCollectStats()
Denable_gradient_descent.cc44 if (!autotune_) { in OptimizeAndCollectStats()
Dbatch_parallelization.cc65 if (!autotune_) { in OptimizeAndCollectStats()
Dmap_parallelization.cc62 if (!autotune_) { in OptimizeAndCollectStats()
Dautotune_buffer_sizes.cc55 if (!autotune_) { in OptimizeAndCollectStats()
/external/tensorflow/tensorflow/core/framework/
Dmodel.h148 autotune_(true), in Node()
198 return autotune_; in autotune()
310 autotune_.store(autotune); in set_autotune()
566 std::atomic<bool> autotune_; variable
Dmodel.cc1383 if (!autotune_ || num_elements_ <= 0) { in CollectTunableParametersHelper()
1397 strings::StrAppend(&result, " autotune=", autotune_.load(), "\n"); in DebugStringHelper()
1429 cloned_current->autotune_.store(autotune_); in SnapshotHelper()
1450 if (!autotune_) { in TotalBufferedBytesHelper()
1472 if (!autotune_) { in TotalMaximumBufferedBytesHelper()
1492 node_proto->set_autotune(autotune_); in ToProto()
1523 node->autotune_.store(node_proto.autotune()); in FromProtoHelper()
/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_map_dataset_op.cc212 autotune_(params.dataset->num_parallel_calls_ == model::kAutotune) {} in Iterator()
364 std::make_pair("autotune", autotune_ ? "true" : "false")); in GetTraceMeMetadata()
643 const bool autotune_; member in tensorflow::data::ParallelMapDatasetOp::Dataset::Iterator
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparse_example_dataset_op.cc372 autotune_(params.dataset->num_parallel_calls_ == model::kAutotune) { in Iterator()
519 std::make_pair("autotune", autotune_ ? "true" : "false")); in GetTraceMeMetadata()
892 const bool autotune_; member in tensorflow::data::experimental::__anonfe9643fa0111::ParseExampleDatasetOp::Dataset::Iterator