Home
last modified time | relevance | path

Searched refs:pin_to_host_optimization (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmeta_optimizer.cc258 new PinToHostOptimizer(cfg_.pin_to_host_optimization())); in MakeNewOptimizer()
401 if (BOTH_ARE_ON(pin_to_host_optimization)) in InitializeOptimizers()
403 else if (BOTH_ARE_EXPERIMENTAL_MLIR(pin_to_host_optimization) || in InitializeOptimizers()
404 BOTH_ARE_EXPERIMENTAL_BOTH(pin_to_host_optimization)) in InitializeOptimizers()
627 PRINT_CFG(pin_to_host_optimization) in PrintUserAndPluginConfigs()
1347 rewrite_cfg.pin_to_host_optimization() == RewriterConfig::ON || in MetaOptimizerEnabled()
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/grappler/
H A Dgrappler.h113 TF_TriState pin_to_host_optimization; member
H A Dgrappler.cc129 CONFIG_TOGGLE(pin_to_host_optimization); in CGraphOptimizerRegister()
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/
H A Dutils.py48 rewriter_config.pin_to_host_optimization = off
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/
H A Ddebug_graph_reconstruction_test.py43 pin_to_host_optimization=rewriter_config_pb2.RewriterConfig.OFF,
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Drewriter_config.proto101 Toggle pin_to_host_optimization = 18; field
/aosp_15_r20/external/tensorflow/tensorflow/python/profiler/
H A Dmodel_analyzer_test.py51 pin_to_host_optimization=rewriter_config_pb2.RewriterConfig.OFF)
/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.h1174 ::tensorflow::RewriterConfig_Toggle pin_to_host_optimization() const; in Swap()
1843 inline ::tensorflow::RewriterConfig_Toggle RewriterConfig::pin_to_host_optimization() const { in pin_to_host_optimization() function
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/cli/
H A Danalyzer_cli_test.py79 pin_to_host_optimization=rewriter_config_pb2.RewriterConfig.OFF)
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dtest_util.py3647 config.graph_options.rewrite_options.pin_to_host_optimization = (