Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dauto_shard_dataset_op.cc89 auto custom_optimizer = rewriter_config.add_custom_optimizers(); in CreateConfig() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmeta_optimizer.cc515 auto custom_optimizer = in InitializeOptimizersByName() local
542 auto custom_optimizer = CustomGraphOptimizerRegistry::CreateByNameOrNull( in InitializeCustomGraphOptimizers() local
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Drewrite_utils.cc146 auto custom_optimizer = rewriter_config.add_custom_optimizers(); in CreateRewriterConfig() local