Home
last modified time | relevance | path

Searched refs:implementation_selector (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dutils.py38 rewriter_config.implementation_selector = off
/external/tensorflow/tensorflow/core/grappler/optimizers/
DBUILD680 ":implementation_selector",
1069 name = "implementation_selector",
1070 srcs = ["implementation_selector.cc"],
1071 hdrs = ["implementation_selector.h"],
1097 ":implementation_selector",
Dmeta_optimizer.cc238 if (cfg_.implementation_selector() != RewriterConfig::OFF) { in InitializeOptimizers()
864 ImplementationSelector implementation_selector; in OptimizeConsumeItem() local
873 TF_RETURN_IF_ERROR(implementation_selector.Optimize( in OptimizeConsumeItem()
/external/tensorflow/tensorflow/core/protobuf/
Drewriter_config.proto97 Toggle implementation_selector = 22; field
/external/tensorflow/tensorflow/python/keras/layers/
Dgru_v2_test.py56 _rewrites.implementation_selector = rewriter_config_pb2.RewriterConfig.ON
Dlstm_v2_test.py57 _rewrites.implementation_selector = rewriter_config_pb2.RewriterConfig.ON
/external/tensorflow/tensorflow/python/eager/
Dfunction_test.py2816 rewrites.implementation_selector = rewriter_config_pb2.RewriterConfig.ON