Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dutils.py47 rewriter_config.pin_to_host_optimization = off
/external/tensorflow/tensorflow/c/experimental/grappler/
Dgrappler.h112 TF_TriState pin_to_host_optimization; member
Dgrappler.cc127 CONFIG_TOGGLE(pin_to_host_optimization); in CGraphOptimizerRegister()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer.cc238 new PinToHostOptimizer(cfg_.pin_to_host_optimization())); in MakeNewOptimizer()
317 if (BOTH_ARE_ON(pin_to_host_optimization)) { in InitializeOptimizers()
500 PRINT_CFG(pin_to_host_optimization) in PrintUserAndPluginConfigs()
1160 rewrite_cfg.pin_to_host_optimization() == RewriterConfig::ON || in MetaOptimizerEnabled()
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_graph_reconstruction_test.py47 pin_to_host_optimization=rewriter_config_pb2.RewriterConfig.OFF,
/external/tensorflow/tensorflow/core/protobuf/
Drewriter_config.proto94 Toggle pin_to_host_optimization = 18; field
/external/tensorflow/tensorflow/python/profiler/
Dmodel_analyzer_test.py55 pin_to_host_optimization=rewriter_config_pb2.RewriterConfig.OFF)
/external/tensorflow/tensorflow/python/debug/cli/
Danalyzer_cli_test.py84 pin_to_host_optimization=rewriter_config_pb2.RewriterConfig.OFF)
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py3377 config.graph_options.rewrite_options.pin_to_host_optimization = (