| /external/tensorflow/tensorflow/compiler/tests/ |
| D | BUILD | 93 enable_mlir_bridge = False,
111 enable_mlir_bridge = False,
130 enable_mlir_bridge = False,
149 enable_mlir_bridge = False,
171 enable_mlir_bridge = False,
190 enable_mlir_bridge = True,
209 enable_mlir_bridge = True,
234 enable_mlir_bridge = True,
260 enable_mlir_bridge = True,
278 enable_mlir_bridge = True,
[all …]
|
| D | build_defs.bzl | 23 enable_mlir_bridge = True,
53 enable_mlir_bridge: If true, then runs the test with and without mlir
106 if enable_mlir_bridge:
|
| D | xla_test.py | 111 context.context().enable_mlir_bridge = True
113 context.context().enable_mlir_bridge = False
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | flags.cc | 229 bool enable_mlir_bridge = false; in AllocateAndParseFlags() local
287 Flag("tf_mlir_enable_mlir_bridge", &enable_mlir_bridge, in AllocateAndParseFlags()
318 } else if (enable_mlir_bridge) { in AllocateAndParseFlags()
423 if (config_proto.value().experimental().enable_mlir_bridge()) { in GetMlirBridgeRolloutState()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config.py | 878 def enable_mlir_bridge(): function
891 context.context().enable_mlir_bridge = True
914 context.context().enable_mlir_bridge = False
|
| D | config_test.py | 209 self.assertFalse(context.context().config.experimental.enable_mlir_bridge)
215 config.enable_mlir_bridge()
216 self.assertTrue(context.context().config.experimental.enable_mlir_bridge)
223 self.assertFalse(context.context().config.experimental.enable_mlir_bridge)
231 self.assertFalse(context.context().config.experimental.enable_mlir_bridge)
|
| D | test_util.py | 2378 context.context().enable_mlir_bridge = True
2380 context.context().enable_mlir_bridge = False
|
| /external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| D | saved_model_testutil.cc | 100 bool enable_mlir_bridge, bool disable_grappler) { in ComputeCurrentTFResult() argument
104 enable_mlir_bridge); in ComputeCurrentTFResult()
142 bool enable_mlir_bridge, bool disable_grappler) { in ComputeCurrentTFResult() argument
145 output_names, outputs, &bundle, enable_mlir_bridge, in ComputeCurrentTFResult()
|
| D | saved_model_testutil.h | 91 bool enable_mlir_bridge = false,
105 bool enable_mlir_bridge = false,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/g3doc/ |
| D | enable_mlir_bridge.md | 14 enable_mlir_bridge=True,
28 tf.config.experimental.enable_mlir_bridge()
|
| /external/tensorflow/tensorflow/python/tpu/tests/ |
| D | tpu_embedding_v2_enqueue_mode_test.py | 35 config.enable_mlir_bridge()
79 config.enable_mlir_bridge()
|
| D | tpu_embedding_v2_invalid_input_test.py | 205 config.enable_mlir_bridge()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-config-proto.-experimental.pbtxt | 73 name: "enable_mlir_bridge"
|
| D | tensorflow.config.experimental.pbtxt | 20 name: "enable_mlir_bridge"
|
| D | tensorflow.-config-proto.pbtxt | 202 name: "enable_mlir_bridge"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.config.experimental.pbtxt | 20 name: "enable_mlir_bridge"
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 1095 config.experimental.enable_mlir_bridge = True
1723 def enable_mlir_bridge(self): member in Context
1730 @enable_mlir_bridge.setter
1731 def enable_mlir_bridge(self, enabled): member in Context
|
| D | BUILD | 803 enable_mlir_bridge = False,
1024 enable_mlir_bridge = True,
1048 enable_mlir_bridge = False,
|
| /external/tensorflow/tensorflow/ |
| D | BUILD | 757 name = "enable_mlir_bridge",
758 define_values = {"enable_mlir_bridge": "true"},
765 define_values = {"enable_mlir_bridge": "false"},
|
| D | tensorflow.bzl | 3282 str(Label("//tensorflow:enable_mlir_bridge")): [
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | config.proto | 644 bool enable_mlir_bridge = 13; field
668 // This field is underdevelopment, for now use enable_mlir_bridge
|
| /external/tensorflow/ |
| D | RELEASE.md | 6225 * Add `tf.config.experimental.enable_mlir_bridge` to allow using MLIR
|