Searched refs:tfcompile (Results 1 – 25 of 33) 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()
73 tensorflow::string usage = tensorflow::tfcompile::kUsageHeader; in main()
86 tensorflow::Status status = tensorflow::tfcompile::Main(flags); in main()
|
| D | tfcompile.bzl | 5 load("//tensorflow/compiler/aot:tfcompile.bzl", "tf_library")
37 tfcompile_tool = "//tensorflow/compiler/aot:tfcompile",
43 """Runs tfcompile to compile a TensorFlow graph into executable code.
82 tfcompile_flags: Extra flags to pass to tfcompile to control compilation.
83 tfcompile_tool: The tfcompile binary. A non-default can be passed to
84 use a tfcompile built with extra dependencies.
109 # First run tfcompile to generate the list of out_nodes.
125 # Run tfcompile on the build host, rather than forge, since it's
164 # Rule that runs tfcompile to produce the header and object file.
224 # Run tfcompile on the build host since it's typically faster on the
[all …]
|
| D | aot_only_var_handle_op.cc | 57 REGISTER_OP(tfcompile::kXlaAotOnlyVarHandleOp)
81 REGISTER_XLA_OP(Name(tfcompile::kXlaAotOnlyVarHandleOp).CompilationOnly(),
|
| D | BUILD | 1 load("//tensorflow/compiler/aot:tfcompile.bzl", "tf_library")
94 name = "tfcompile",
134 # tfcompile.bzl correctly handles usage from outside of the package that it is
191 # added by the tfcompile bazel macro.
|
| D | aot_only_var_handle_op.h | 19 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 | 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.
|
| D | benchmark.cc | 34 namespace tfcompile { namespace
|
| D | embedded_protocol_buffers.cc | 35 namespace tfcompile { namespace
|
| /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 …]
|
| D | index.md | 100 ### AOT (Ahead-of-time) compilation for CPU with `tfcompile`
102 You can also use a standalone [`tfcompile`](./tfcompile) tool,
|
| /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/compiler/mlir/xla/tests/translate/ |
| D | types.hlotxt | 3 HloModule tfcompile.1
6 ENTRY %tfcompile.1 {
|
| D | conditional.hlotxt | 3 HloModule tfcompile.20
20 ENTRY %tfcompile.20 {
|
| D | fully_connected_reference_model.hlotxt | 5 HloModule tfcompile.48
8 ENTRY %tfcompile.48 {
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla.cc | 133 node.set_op(tfcompile::kXlaAotOnlyVarHandleOp); in ConvertVarHandlesToAotVarHandles()
139 node.set_op(tfcompile::kXlaAotOnlyVarHandleOp); in ConvertVarHandlesToAotVarHandles()
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | BUILD | 1 load("//tensorflow/compiler/aot:tfcompile.bzl", "tf_library")
9 # We disable some tfcompile tests in the open source build with the
99 # expand the "required by all tf_library targets" list in tfcompile.bzl.
|
12