tf2xla_util.h File Reference

#include <unordered_map>
#include "tensorflow/compiler/tf2xla/tf2xla.pb.h"
#include "tensorflow/core/framework/graph.pb.h"
#include "tensorflow/core/framework/op.h"
#include "tensorflow/core/graph/graph.h"
#include "tensorflow/core/lib/core/status.h"

Include dependency graph for tf2xla_util.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
	tensorflow

Functions
Status	tensorflow::ValidateConfig (const tf2xla::Config &config)
Status	tensorflow::AddPlaceholdersForFeeds (const tf2xla::Config &config, const OpRegistryInterface *op_registry, std::unordered_map< string, string > *feed_remapping, GraphDef *graph_def)
	Generate corresponding placeholder for each feed tensor. Traverse each inputs of each nodes. Replace feeds to corresponding placeholders. More...
Status	tensorflow::PruneGraphDefInto (const tf2xla::Config &config, const GraphDef &in, GraphDef *out)
	Remove source nodes of feed tensor. Because they are removed at tensorflow::AddPlaceholdersForFeeds More...