Home
last modified time | relevance | path

Searched refs:scoped_allocator_optimization (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmeta_optimizer.cc255 new ScopedAllocatorOptimizer(cfg_.scoped_allocator_optimization(), in MakeNewOptimizer()
482 if (BOTH_ARE_ON(scoped_allocator_optimization)) { in InitializeOptimizers()
484 cfg_.scoped_allocator_optimization(), cfg_.scoped_allocator_opts())); in InitializeOptimizers()
485 } else if (BOTH_ARE_EXPERIMENTAL_MLIR(scoped_allocator_optimization) || in InitializeOptimizers()
486 BOTH_ARE_EXPERIMENTAL_BOTH(scoped_allocator_optimization)) { in InitializeOptimizers()
632 PRINT_CFG(scoped_allocator_optimization) in PrintUserAndPluginConfigs()
1345 rewrite_cfg.scoped_allocator_optimization() == RewriterConfig::ON || in MetaOptimizerEnabled()
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/grappler/
H A Dgrappler.h120 TF_TriState scoped_allocator_optimization; member
124 TF_OFFSET_OF_END(TP_OptimizerConfigs, scoped_allocator_optimization)
H A Dgrappler.cc136 CONFIG_TOGGLE(scoped_allocator_optimization); in CGraphOptimizerRegister()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcollective_ops_test.py186 rewrite_options.scoped_allocator_optimization = (
247 rewrite_options.scoped_allocator_optimization = (
541 rewrite_options.scoped_allocator_optimization = (
H A Dcollective_ops_xla_test.py42 rewrite_options.scoped_allocator_optimization = (
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/
H A Dutils.py50 rewriter_config.scoped_allocator_optimization = off
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Drewriter_config.proto99 Toggle scoped_allocator_optimization = 15; field
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dcollective_all_reduce_strategy.py752 rewrite_options.scoped_allocator_optimization = (
H A Dcollective_all_reduce_strategy_test.py420 new_rewrite_options.scoped_allocator_optimization)
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dconfig_test.py855 new_rewrite_options.scoped_allocator_optimization)
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Drewriter_config.pb.h1156 ::tensorflow::RewriterConfig_Toggle scoped_allocator_optimization() const; in Swap()
1823 inline ::tensorflow::RewriterConfig_Toggle RewriterConfig::scoped_allocator_optimization() const { in scoped_allocator_optimization() function
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dcontext.py1161 rewrite_options.scoped_allocator_optimization = (