tf2xla_util.cc File Reference

#include "tensorflow/compiler/tf2xla/tf2xla_util.h"
#include <queue>
#include <set>
#include <unordered_map>
#include "tensorflow/compiler/tf2xla/sharding_util.h"
#include "tensorflow/compiler/tf2xla/tf2xla.pb.h"
#include "tensorflow/compiler/xla/xla_data.pb.h"
#include "tensorflow/core/framework/graph.pb.h"
#include "tensorflow/core/framework/graph_def_util.h"
#include "tensorflow/core/framework/node_def.pb.h"
#include "tensorflow/core/framework/tensor_shape.h"
#include "tensorflow/core/framework/tensor_shape.pb.h"
#include "tensorflow/core/framework/versions.pb.h"
#include "tensorflow/core/graph/tensor_id.h"
#include "tensorflow/core/lib/core/errors.h"
#include "tensorflow/core/lib/core/status.h"
#include "tensorflow/core/lib/gtl/optional.h"
#include "tensorflow/core/lib/strings/strcat.h"

Include dependency graph for tf2xla_util.cc:

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...