Searched refs:mlir_components (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | BUILD | 307 mlir_components = "Bridge",
320 mlir_components = "Bridge",
334 mlir_components = "Bridge",
346 mlir_components = "Bridge",
358 mlir_components = "Bridge",
370 mlir_components = "Bridge",
382 mlir_components = "Bridge",
394 mlir_components = "Bridge",
408 mlir_components = "Bridge",
420 mlir_components = "Bridge",
[all …]
|
| /external/tensorflow/tensorflow/python/ |
| D | tfcompile_wrapper.cc | 42 std::string mlir_components, bool gen_name_to_index, in PYBIND11_MODULE()
56 flags.mlir_components = std::move(mlir_components); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | compile.cc | 107 if (flags.mlir_components == "Bridge") { in CompileGraph()
111 if (!flags.mlir_components.empty()) { in CompileGraph()
112 return errors::Unknown("Unknown mlir_components ", flags.mlir_components); in CompileGraph()
|
| D | flags.h | 42 string mlir_components; member
|
| D | flags.cc | 71 {"mlir_components", &flags->mlir_components, in AppendMainFlags()
|
| D | tfcompile.bzl | 40 mlir_components = None,
91 mlir_components: When the value is "Bridge", use MLIR to translate
192 if mlir_components:
193 mlir_flag = "--mlir_components=" + mlir_components
|