|
tf_1.8_xla_doc
|
#include "tensorflow/compiler/tf2xla/tf2xla.h"#include <map>#include <memory>#include <string>#include <unordered_map>#include <utility>#include <vector>#include "tensorflow/compiler/tf2xla/dump_graph.h"#include "tensorflow/compiler/tf2xla/shape_util.h"#include "tensorflow/compiler/tf2xla/tf2xla_util.h"#include "tensorflow/compiler/tf2xla/xla_compiler.h"#include "tensorflow/compiler/tf2xla/xla_op_registry.h"#include "tensorflow/core/common_runtime/function.h"#include "tensorflow/core/framework/function.h"#include "tensorflow/core/framework/graph.pb.h"#include "tensorflow/core/framework/graph_def_util.h"#include "tensorflow/core/framework/op.h"#include "tensorflow/core/framework/tensor_shape.h"#include "tensorflow/core/framework/versions.pb.h"#include "tensorflow/core/graph/algorithm.h"#include "tensorflow/core/graph/graph.h"#include "tensorflow/core/graph/graph_constructor.h"#include "tensorflow/core/graph/node_builder.h"#include "tensorflow/core/lib/core/errors.h"#include "tensorflow/core/lib/strings/str_util.h"#include "tensorflow/core/lib/strings/strcat.h"#include "tensorflow/core/platform/logging.h"#include "tensorflow/core/platform/types.h"Namespaces | |
| tensorflow | |
Functions | |
| Status | tensorflow::anonymous_namespace{tf2xla.cc}::RewriteAndPruneGraph (Graph *graph, const tf2xla::Config &config, const std::unordered_map< string, string > &feed_remapping) |
| Status | tensorflow::anonymous_namespace{tf2xla.cc}::ConvertGraphToXla (std::unique_ptr< Graph > graph, xla::Client *client, xla::Computation *computation) |
| Converts the TensorFlow graph into an XLA computation, by executing the graph symbolically, with each op building up the XLA HLO. More... | |
| Status | tensorflow::anonymous_namespace{tf2xla.cc}::InitGraph (const GraphDef &graph_def, const tf2xla::Config &config, std::unique_ptr< Graph > *graph) |
| InitGraph creates a graph based on the graph_def, that may then be convert to an xla::Computation via ConvertGraphToXla. More... | |
| Status | tensorflow::ConvertGraphDefToXla (const GraphDef &graph_def, const tf2xla::Config &config, xla::Client *client, xla::Computation *computation) |
| Convert tensorflow graph_def to XLA UserComputation. More... | |
1.8.14