blob: f1f38468759b5ba6bf1deb57e56cd31e6ddc7c55 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
#ifndef TENSORFLOW_GRAPH_TENSOR_ID_H_
#define TENSORFLOW_GRAPH_TENSOR_ID_H_
#include <string>
#include "tensorflow/core/lib/strings/strcat.h"
#include "tensorflow/core/lib/core/stringpiece.h"
namespace tensorflow {
// Identifier for a tensor within a step.
// first == operation_name, second == output_index
// Note: does not own backing storage for name.
struct TensorId : public std::pair<StringPiece, int> {
typedef std::pair<StringPiece, int> Base;
// Inherit the set of constructors.
using Base::pair;
string ToString() const { return strings::StrCat(first, ":", second); }
};
TensorId ParseTensorName(const string& name);
TensorId ParseTensorName(StringPiece name);
} // namespace tensorflow
#endif // TENSORFLOW_GRAPH_TENSOR_ID_H_
|