Searched refs:autotune_algorithm (Results 1 – 7 of 7) sorted by relevance
65 params->autotune_algorithm = model::AutotuneAlgorithm::DEFAULT; in SetRootDatasetParams()67 params->autotune_algorithm = model::AutotuneAlgorithm::STAGE_BASED; in SetRootDatasetParams()71 params->autotune_algorithm = in SetRootDatasetParams()72 options.autotune_options().autotune_algorithm(); in SetRootDatasetParams()86 kAlgorithm, model::AutotuneAlgorithm_Name(params.autotune_algorithm))); in AddTraceMetadata()261 model_->OptimizeLoop(dataset()->params_.autotune_algorithm, in EnsureModelThreadStarted()
32 model::AutotuneAlgorithm autotune_algorithm; member
224 autotune_algorithm = options_lib.create_option( variable in AutotuneOptions238 if self.autotune_algorithm is not None:239 pb.autotune_algorithm = AutotuneAlgorithm._to_proto( # pylint: disable=protected-access240 self.autotune_algorithm)251 self.autotune_algorithm = AutotuneAlgorithm._from_proto( # pylint: disable=protected-access252 pb.autotune_algorithm)
7 name: "autotune_algorithm"
55 model.AutotuneAlgorithm autotune_algorithm = 4; field
86 …* Added a new value to `tf.data.Options.autotune.autotune_algorithm`: STAGE_BASED. If the autotu…5815 * Removed `autotune_algorithm` from experimental optimization options.