Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_interleave_dataset_op_test.cc137 /*buffer_output_elements=*/model::kAutotune, in ParallelInterleaveDatasetParams1()
138 /*prefetch_input_elements=*/model::kAutotune, in ParallelInterleaveDatasetParams1()
287 /*buffer_output_elements=*/model::kAutotune, in ParallelInterleaveDatasetParams7()
288 /*prefetch_input_elements=*/model::kAutotune, in ParallelInterleaveDatasetParams7()
312 /*buffer_output_elements=*/model::kAutotune, in ParallelInterleaveDatasetParams8()
313 /*prefetch_input_elements=*/model::kAutotune, in ParallelInterleaveDatasetParams8()
337 /*buffer_output_elements=*/model::kAutotune, in ParallelInterleaveDatasetParams9()
338 /*prefetch_input_elements=*/model::kAutotune, in ParallelInterleaveDatasetParams9()
362 /*buffer_output_elements=*/model::kAutotune, in ParallelInterleaveDatasetParams10()
363 /*prefetch_input_elements=*/model::kAutotune, in ParallelInterleaveDatasetParams10()
[all …]
Dprefetch_autotuner_test.cc36 PrefetchAutotuner t(model::kAutotune, 0); in TEST()
61 PrefetchAutotuner t(model::kAutotune, 0); in TEST()
84 PrefetchAutotuner t(model::kAutotune, 2); in TEST()
Dparallel_interleave_dataset_op.cc127 if (configured_buffer_output_elements != model::kAutotune) { in ComputeBufferOutputElements()
135 if (configured_prefetch_input_elements != model::kAutotune) { in ComputePrefetchInputElements()
183 num_parallel_calls == model::kAutotune ? "true" : "false"}, in Dataset()
344 if (num_parallel_calls_->value == model::kAutotune) { in Initialize()
1604 int64 buffer_output_elements = model::kAutotune; in MakeDataset()
1605 int64 prefetch_input_elements = model::kAutotune; in MakeDataset()
1610 buffer_output_elements == model::kAutotune || in MakeDataset()
1613 model::kAutotune, " or > 0 but is ", in MakeDataset()
1619 prefetch_input_elements == model::kAutotune || in MakeDataset()
1622 model::kAutotune, " or >= 0 but is ", in MakeDataset()
[all …]
Dprefetch_dataset_op.cc157 if (buffer_size_->value == model::kAutotune) { in Initialize()
327 dataset()->buffer_size_ == model::kAutotune ? "true" : "false")); in GetTraceMeMetadata()
611 OP_REQUIRES(ctx, buffer_size >= 0 || buffer_size == model::kAutotune, in MakeDataset()
614 model::kAutotune, " for auto-tuning")); in MakeDataset()
616 if (buffer_size == model::kAutotune) { in MakeDataset()
Dprefetch_autotuner.cc26 if (initial_buffer_size == model::kAutotune) { in PrefetchAutotuner()
Dparallel_map_dataset_op_test.cc186 /*num_parallel_calls=*/model::kAutotune, in ParallelMapDatasetParams5()
243 /*num_parallel_calls=*/model::kAutotune, in ParallelMapDatasetParams8()
Dparallel_map_dataset_op.cc212 autotune_(params.dataset->num_parallel_calls_ == model::kAutotune) {} in Iterator()
223 if (num_parallel_calls_->value == model::kAutotune) { in Initialize()
718 ctx, num_parallel_calls > 0 || num_parallel_calls == model::kAutotune, in MakeDataset()
726 if (num_parallel_calls == model::kAutotune) { in MakeDataset()
Dparallel_batch_dataset_op.cc81 num_parallel_calls == model::kAutotune ? "true" : "false"}, in Dataset()
183 if (num_parallel_calls_->value == model::kAutotune) { in Initialize()
Dinterleave_dataset_op.cc324 if (cycle_length == model::kAutotune) { in MakeDataset()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Denable_gradient_descent.h25 constexpr char kAutotune[] = "autotune"; variable
41 const string& autotune = config->parameter_map().at(kAutotune).s(); in Init()
48 kAutotune, ": ", autotune); in Init()
Dbatch_parallelization.h25 constexpr char kAutotune[] = "autotune"; variable
41 const string& autotune = config->parameter_map().at(kAutotune).s(); in Init()
48 kAutotune, ": ", autotune); in Init()
Ddisable_prefetch_legacy_autotune.h25 constexpr char kAutotune[] = "autotune"; variable
41 const string& autotune = config->parameter_map().at(kAutotune).s(); in Init()
48 kAutotune, ": ", autotune); in Init()
Dautotune_buffer_sizes.h25 constexpr char kAutotune[] = "autotune"; variable
49 const string& autotune = config->parameter_map().at(kAutotune).s(); in Init()
56 kAutotune, ": ", autotune); in Init()
Dmap_parallelization.h25 constexpr char kAutotune[] = "autotune"; variable
41 const string& autotune = config->parameter_map().at(kAutotune).s(); in Init()
48 kAutotune, ": ", autotune); in Init()
Dautotune_buffer_sizes.cc64 graph_utils::AddScalarConstNode(data::model::kAutotune, &graph); in OptimizeAndCollectStats()
77 if (initial_buffer_size != data::model::kAutotune) { in OptimizeAndCollectStats()
Dbatch_parallelization.cc49 graph_utils::AddScalarConstNode(data::model::kAutotune, graph); in MakeParallelBatch()
Dmap_parallelization.cc48 static_cast<int64>(data::model::kAutotune), graph); in MakeParallelMap()
Dmap_vectorization.cc285 static_cast<int32>(data::model::kAutotune), graph); in AddNewMapNode()
289 graph_utils::AddScalarConstNode(data::model::kAutotune, graph); in AddNewMapNode()
329 static_cast<int64>(data::model::kAutotune), graph); in AddNewPrefetchNode()
/external/tensorflow/tensorflow/core/framework/
Dmodel_test.cc522 /*value=*/model::kAutotune, nullptr, nullptr), in TEST()
559 /*value=*/model::kAutotune, nullptr, nullptr), in TEST()
578 /*value=*/model::kAutotune, nullptr, nullptr), in TEST()
601 std::make_shared<SharedState>(/*value=*/model::kAutotune, nullptr, in TEST()
614 /*value=*/model::kAutotune, nullptr, nullptr), in TEST()
671 /*value=*/model::kAutotune, nullptr, nullptr), in TEST_P()
678 /*value=*/model::kAutotune, nullptr, nullptr), in TEST_P()
728 /*value=*/model::kAutotune, nullptr, nullptr), in TEST()
762 /*value=*/model::kAutotune, nullptr, nullptr), in TEST()
793 /*value=*/model::kAutotune, nullptr, nullptr), in TEST()
[all …]
Dmodel.h45 constexpr int64 kAutotune = -1; variable
66 tunable(value == kAutotune) {} in SharedState()
86 value(state->value == kAutotune ? min : state->value), in Parameter()
Dmodel.cc1540 std::make_shared<SharedState>(kAutotune, std::make_shared<mutex>(), in FromProtoHelper()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op.cc101 num_parallel_calls == model::kAutotune ? "true" : "false"}, in Dataset()
205 CeilDiv(params.dataset->num_parallel_calls_ == model::kAutotune in Iterator()
218 if (num_parallel_calls_->value == model::kAutotune) { in Initialize()
712 ctx, num_parallel_calls > 0 || num_parallel_calls == model::kAutotune, in MakeDataset()
724 if (num_parallel_calls == model::kAutotune) { in MakeDataset()
Ddata_service_dataset_op.cc548 if (dataset()->max_outstanding_requests_ == model::kAutotune) { in UpdateTasks()
860 if (task_refresh_interval_hint_ms_ == model::kAutotune) { in DataServiceDatasetOp()
959 max_outstanding_requests == model::kAutotune || in MakeDataset()
962 model::kAutotune)); in MakeDataset()
Dmap_and_batch_dataset_op_test.cc183 /*num_parallel_calls=*/model::kAutotune, in MapAndBatchDatasetParams5()
Dparse_example_dataset_op.cc125 ctx, num_parallel_calls > 0 || num_parallel_calls == model::kAutotune, in MakeDataset()
372 autotune_(params.dataset->num_parallel_calls_ == model::kAutotune) { in Iterator()
382 if (num_parallel_calls_->value == model::kAutotune) { in Initialize()