| /external/tensorflow/tensorflow/cc/saved_model/experimental/tests/ |
| D | saved_model_api_test.cc | 52 bool use_tfrt = GetParam(); in TEST_P() local
53 if (use_tfrt) { in TEST_P()
57 builder.SetUseTFRT(use_tfrt); in TEST_P()
76 bool use_tfrt = GetParam(); in TEST_P() local
77 if (use_tfrt) { in TEST_P()
81 builder.SetUseTFRT(use_tfrt); in TEST_P()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
| D | saved_model_api_test.cc | 62 bool use_tfrt = GetParam(); in TEST_P() local
63 if (use_tfrt) { in TEST_P()
69 TFE_ContextOptionsSetTfrt(opts, use_tfrt); in TEST_P()
94 bool use_tfrt = GetParam(); in TEST_P() local
95 if (use_tfrt) { in TEST_P()
101 TFE_ContextOptionsSetTfrt(opts, use_tfrt); in TEST_P()
192 bool use_tfrt = GetParam(); in TEST_P() local
193 if (use_tfrt) { in TEST_P()
199 TFE_ContextOptionsSetTfrt(opts, use_tfrt); in TEST_P()
296 bool use_tfrt = GetParam(); in TEST_P() local
[all …]
|
| /external/tensorflow/tensorflow/cc/experimental/base/public/ |
| D | runtime_builder.h | 39 RuntimeBuilder& SetUseTFRT(bool use_tfrt);
67 inline RuntimeBuilder& RuntimeBuilder::SetUseTFRT(bool use_tfrt) { in SetUseTFRT() argument
68 TFE_ContextOptionsSetTfrt(options_.get(), use_tfrt); in SetUseTFRT()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_internal.h | 36 bool use_tfrt = false; member
|
| D | unified_api_testutil.h | 50 Status BuildImmediateExecutionContext(bool use_tfrt, AbstractContext** ctx);
|
| D | unified_api_testutil.cc | 122 Status BuildImmediateExecutionContext(bool use_tfrt, AbstractContext** ctx) { in BuildImmediateExecutionContext() argument
126 TFE_ContextOptionsSetTfrt(opts, use_tfrt); in BuildImmediateExecutionContext()
|
| D | gradients_util.h | 85 Status BuildImmediateExecutionContext(bool use_tfrt, AbstractContext** ctx);
|
| D | c_api_experimental.cc | 485 void TFE_ContextOptionsSetTfrt(TFE_ContextOptions* options, bool use_tfrt) { in TFE_ContextOptionsSetTfrt() argument
486 options->use_tfrt = use_tfrt; in TFE_ContextOptionsSetTfrt()
|
| D | gradients_util.cc | 308 Status BuildImmediateExecutionContext(bool use_tfrt, AbstractContext** ctx) { in BuildImmediateExecutionContext() argument
312 TFE_ContextOptionsSetTfrt(opts, use_tfrt); in BuildImmediateExecutionContext()
|
| D | c_api_experimental.h | 270 bool use_tfrt);
|
| D | c_api_test.cc | 1054 void RunAddFunction(bool use_tfrt, bool enable_grappler) { in RunAddFunction() argument
1057 TFE_ContextOptionsSetTfrt(opts, use_tfrt); in RunAddFunction()
1089 if (use_tfrt) { in RunAddFunction()
1120 if (use_tfrt) { in RunAddFunction()
|
| D | c_api.cc | 116 if (opts->use_tfrt) { in TFE_NewContext()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-config-proto.-experimental.pbtxt | 104 name: "use_tfrt"
|
| D | tensorflow.-config-proto.pbtxt | 233 name: "use_tfrt"
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 1690 def use_tfrt(self): member in Context
1693 @use_tfrt.setter
1694 def use_tfrt(self, tfrt): member in Context
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | replicate_test.py | 258 if context.context().use_tfrt:
|
| D | snapshot_test.py | 390 if context.context().use_tfrt:
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 178 if (config_proto_.experimental().use_tfrt()) return false; in LowerControlFlow()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | config.proto | 630 bool use_tfrt = 18; field
|