Searched refs:tfcompile (Results 1 – 25 of 39) sorted by relevance
12
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | benchmark_main.template | 4 // generated by tfcompile. All tokens of the form `{{TFCOMPILE_*}}` must be
7 // TFCOMPILE_HEADER : Path to the header file generated by tfcompile.
8 // TFCOMPILE_CPP_CLASS : Name of the C++ class generated by tfcompile.
10 // The tf_library bazel macro in tfcompile.bzl performs the token rewriting, and
30 namespace tfcompile {
46 } // namespace tfcompile
50 return tensorflow::tfcompile::Main(argc, argv);
|
| D | tfcompile_main.cc | 46 namespace tfcompile { namespace
62 tensorflow::tfcompile::MainFlags flags; in main()
76 tensorflow::string usage = tensorflow::tfcompile::kUsageHeader; in main()
88 tensorflow::Status status = tensorflow::tfcompile::Main(flags); in main()
|
| D | tfcompile.bzl | 5 load("//tensorflow/compiler/aot:tfcompile.bzl", "tf_library")
38 tfcompile_tool = "//tensorflow/compiler/aot:tfcompile",
45 """Runs tfcompile to compile a TensorFlow graph into executable code with fast
85 tfcompile_flags: Extra flags to pass to tfcompile to control compilation.
86 tfcompile_tool: The tfcompile binary. A non-default can be passed to
87 use a tfcompile built with extra dependencies.
94 enable_tracemes: Tell tfcompile to generate calls to
115 # First run tfcompile to generate the list of out_nodes.
131 # Run tfcompile on the build host, rather than forge, since it's
170 # Rule that runs tfcompile to produce the header and object file.
[all …]
|
| D | aot_only_var_handle_op.cc | 57 REGISTER_OP(tfcompile::kXlaAotOnlyVarHandleOp)
81 REGISTER_XLA_OP(Name(tfcompile::kXlaAotOnlyVarHandleOp).CompilationOnly(),
|
| D | aot_only_var_handle_op.h | 19 namespace tfcompile {
|
| D | quantize.h | 29 namespace tfcompile {
|
| D | compile.h | 29 namespace tfcompile {
|
| D | benchmark_test.cc | 22 namespace tfcompile { namespace
|
| D | benchmark.h | 34 namespace tfcompile {
|
| D | embedded_protocol_buffers.h | 28 namespace tfcompile {
|
| D | flags.h | 25 namespace tfcompile {
|
| D | BUILD | 2 load("//tensorflow/compiler/aot:tfcompile.bzl", "tf_library")
109 name = "tfcompile",
151 # tfcompile.bzl correctly handles usage from outside of the package that it is
262 # added by the tfcompile bazel macro.
|
| D | flags.cc | 19 namespace tfcompile { namespace
|
| D | codegen.h | 27 namespace tfcompile {
|
| D | test.cc | 51 namespace tfcompile { namespace
|
| D | test_graph_tfadd.pbtxt | 40 # by tfcompile.
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | test_error_message.lit.pbtxt | 1 # RUN: not tfcompile --graph=%s --config=%s.config.pbtxt --mlir_components=Bridge --debug_info=%s.d…
2 # RUN: not tfcompile --graph=%s --config=%s.config.pbtxt --mlir_components=None 2>&1 | FileCheck -c…
4 # Checks the error message produced by tfcompile with mlir_component
11 # Checks the error message produced by tfcompile without mlir_component
|
| D | BUILD | 6 load("//tensorflow/compiler/aot:tfcompile.bzl", "tf_library")
34 "//tensorflow/compiler/aot:tfcompile",
40 # We disable some tfcompile tests in the open source build with the
132 # expand the "required by all tf_library targets" list in tfcompile.bzl.
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | tfcompile.md | 3 ## What is tfcompile?
5 `tfcompile` is a standalone tool that ahead-of-time (AOT) compiles TensorFlow
7 some runtime overheads. A typical use-case of `tfcompile` is to compile an
14 by `tfcompile` does not use the TensorFlow runtime, and only has dependencies on
21 ## What does tfcompile do?
23 `tfcompile` takes a subgraph, identified by the TensorFlow concepts of
34 ## Using tfcompile
37 `tfcompile` from a TensorFlow subgraph. The steps are:
83 code. `tf_library` utilizes `tfcompile` to compile the TensorFlow graph into
87 load("//tensorflow/compiler/aot:tfcompile.bzl", "tf_library")
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | types.hlotxt | 3 HloModule tfcompile.1
6 ENTRY %tfcompile.1 {
|
| D | if_conditional.hlotxt | 3 HloModule tfcompile.20
20 ENTRY %tfcompile.20 {
|
| D | fully_connected_reference_model.hlotxt | 5 HloModule tfcompile.48
10 ENTRY %tfcompile.48 {
|
| /external/llvm-project/llvm/cmake/modules/ |
| D | TensorFlowCompile.cmake | 7 function(tfcompile model tag_set signature_def_key fname cpp_class) function
28 # Aggregate the objects so that results of different tfcompile calls may be
|
| /external/tensorflow/tensorflow/python/ |
| D | tfcompile_wrapper.cc | 44 tensorflow::tfcompile::MainFlags flags; in PYBIND11_MODULE()
62 tensorflow::MaybeRaiseFromStatus(tensorflow::tfcompile::Main(flags)); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/tools/def_file_filter/ |
| D | symbols_pybind.txt | 342 [//tensorflow/compiler/xla:status_macros] # tfcompile
348 [//tensorflow/compiler/xla/service:hlo] # tfcompile
358 [//tensorflow/compiler/aot:tfcompile_lib] # tfcompile
359 tensorflow::tfcompile::Main
|
12