Home
last modified time | relevance | path

Searched defs:custom_optimizer (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dauto_shard_dataset_op.cc81 auto custom_optimizer = rewriter_config.add_custom_optimizers(); in CreateConfig() local
/external/tensorflow/tensorflow/core/kernels/data/
Doptimize_dataset_op.cc157 auto custom_optimizer = rewriter_config.add_custom_optimizers(); in CreateConfig() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer.cc329 auto custom_optimizer = in InitializeOptimizersByName() local
355 auto custom_optimizer = CustomGraphOptimizerRegistry::CreateByNameOrNull( in InitializeCustomGraphOptimizers() local