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.cc89 auto custom_optimizer = rewriter_config.add_custom_optimizers(); in CreateConfig() local
/external/tensorflow/tensorflow/core/data/
Drewrite_utils.cc146 auto custom_optimizer = rewriter_config.add_custom_optimizers(); in CreateRewriterConfig() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer.cc515 auto custom_optimizer = in InitializeOptimizersByName() local
542 auto custom_optimizer = CustomGraphOptimizerRegistry::CreateByNameOrNull( in InitializeCustomGraphOptimizers() local