tfinfo::v2::TensorSignature Class Reference

Class to store input and output tensor information. More...

#include <TensorSignature.h>

Public Types
enum class	Kind { Input , Output }

Public Member Functions
	TensorSignature (const Kind kind, const std::string &name)
	TensorSignature (const Kind kind, const std::string &name, const ShapeHint &shape_hint)
Kind	kind () const
const TensorName &	name ()
ShapeHint &	shapeHint ()

Detailed Description

Class to store input and output tensor information.

Definition at line 80 of file TensorSignature.h.

Member Enumeration Documentation

Kind

enum class tfinfo::v2::TensorSignature::Kind

strong

Enumerator
Input
Output

Definition at line 83 of file TensorSignature.h.

  {
    Input,
    Output
  };

Constructor & Destructor Documentation

TensorSignature() [1/2]

tfinfo::v2::TensorSignature::TensorSignature ( const Kind  kind, const std::string &  name  )

inline

Definition at line 89 of file TensorSignature.h.

                                                          : _kind(kind), _tensor_name()
  {
    // tensor name can be a form of "placeholder:0" or "placeholder".
    // If tensor index is omitted, ":0" is appended
    auto pos = name.find(":");
    if (pos == std::string::npos)
      _tensor_name.assign(name + ":0");
    else
      _tensor_name.assign(name);
  }

References name().

TensorSignature() [2/2]

tfinfo::v2::TensorSignature::TensorSignature ( const Kind  kind, const std::string &  name, const ShapeHint &  shape_hint  )

inline

Definition at line 100 of file TensorSignature.h.

    : TensorSignature(kind, name)
  {
    _shape_hint = shape_hint;
  }

Member Function Documentation

kind()

Kind tfinfo::v2::TensorSignature::kind ( ) const

inline

Definition at line 107 of file TensorSignature.h.

{ return _kind; }

name()

const TensorName & tfinfo::v2::TensorSignature::name ( )

inline

Definition at line 109 of file TensorSignature.h.

{ return _tensor_name; }

Referenced by TensorSignature().

shapeHint()

ShapeHint & tfinfo::v2::TensorSignature::shapeHint ( )

inline

Definition at line 111 of file TensorSignature.h.

{ return _shape_hint; }

---

The documentation for this class was generated from the following file:

• compiler/tfinfo-v2/include/tfinfo-v2/TensorSignature.h