Home
last modified time | relevance | path

Searched refs:memory_optimization (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/grappler/
H A Dmemory_optimizer_test.py56 memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL))
84 memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL,
144 memory_optimization=(
172 memory_optimization=rewriter_config_pb2.RewriterConfig
201 memory_optimization=rewriter_config_pb2.RewriterConfig
222 memory_optimization=rewriter_config_pb2.RewriterConfig.HEURISTICS)
294 memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL)
315 memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL))
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmeta_optimizer.cc463 if (MemoryOptimizerEnabled(cfg_.memory_optimization(), in InitializeOptimizers()
465 PLUGIN_NOT_OFF(memory_optimization)) { in InitializeOptimizers()
469 MakeUnique<MemoryOptimizer>(cfg_.memory_optimization())); in InitializeOptimizers()
472 cfg_.memory_optimization(), in InitializeOptimizers()
651 MemoryOptimizerEnabled(cfg_.memory_optimization(), in PrintUserAndPluginConfigs()
1342 rewrite_cfg.memory_optimization() != RewriterConfig::NO_MEM_OPT || in MetaOptimizerEnabled()
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/grappler/
H A Dgrappler.h119 TF_TriState memory_optimization; member
H A Dgrappler.cc135 CONFIG_TOGGLE(memory_optimization); in CGraphOptimizerRegister()
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Drewriter_config.proto184 MemOptType memory_optimization = 4; field
188 // heuristics depending on memory_optimization), but the nodes themselves will
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/
H A Dutils.py45 rewriter_config.memory_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.h1075 ::tensorflow::RewriterConfig_MemOptType memory_optimization() const; in Swap()
2103 inline ::tensorflow::RewriterConfig_MemOptType RewriterConfig::memory_optimization() const { in memory_optimization() function
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
H A Dwhile_v2_test.py2077 memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL))