20.02
|
#include <TfParser.hpp>
Public Member Functions | |
virtual armnn::INetworkPtr | CreateNetworkFromTextFile (const char *graphFile, const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs) override |
Creates the network from a protobuf text file on the disk. More... | |
virtual armnn::INetworkPtr | CreateNetworkFromBinaryFile (const char *graphFile, const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs) override |
Creates the network from a protobuf binary file on the disk. More... | |
virtual armnn::INetworkPtr | CreateNetworkFromString (const char *protoText, const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs) override |
Creates the network directly from protobuf text in a string. Useful for debugging/testing. More... | |
virtual BindingPointInfo | GetNetworkInputBindingInfo (const std::string &name) const override |
Retrieves binding info (layer id and tensor info) for the network input identified by the given layer name. More... | |
virtual BindingPointInfo | GetNetworkOutputBindingInfo (const std::string &name) const override |
Retrieves binding info (layer id and tensor info) for the network output identified by the given layer name. More... | |
TfParser () | |
Friends | |
template<typename T > | |
class | ParsedConstTfOperation |
class | ParsedMatMulTfOperation |
class | ParsedMulTfOperation |
Additional Inherited Members | |
Static Public Member Functions inherited from ITfParser | |
static ITfParser * | CreateRaw () |
static ITfParserPtr | Create () |
static void | Destroy (ITfParser *parser) |
Protected Member Functions inherited from ITfParser | |
virtual | ~ITfParser () |
Definition at line 64 of file TfParser.hpp.
TfParser | ( | ) |
Definition at line 515 of file TfParser.cpp.
References INetwork::AddAdditionLayer(), CHECK_LOCATION, IOutputSlot::Connect(), IConnectableLayer::GetInputSlot(), TensorShape::GetNumDimensions(), TensorInfo::GetNumDimensions(), TensorInfo::GetNumElements(), IConnectableLayer::GetOutputSlot(), TensorInfo::GetShape(), IOutputSlot::GetTensorInfo(), armnn::IgnoreUnused(), WithOutputTensorIndex< T >::m_Index, WithOutputTensorIndex< T >::m_IndexedValue, TfParser::ParsedConstTfOperation, TensorInfo::SetShape(), and IOutputSlot::SetTensorInfo().
|
overridevirtual |
Creates the network from a protobuf binary file on the disk.
Implements ITfParser.
Definition at line 3615 of file TfParser.cpp.
References CHECK_LOCATION.
|
overridevirtual |
Creates the network directly from protobuf text in a string. Useful for debugging/testing.
Implements ITfParser.
Definition at line 3595 of file TfParser.cpp.
References CHECK_LOCATION.
|
overridevirtual |
Creates the network from a protobuf text file on the disk.
Implements ITfParser.
Definition at line 3560 of file TfParser.cpp.
References CHECK_LOCATION.
|
overridevirtual |
Retrieves binding info (layer id and tensor info) for the network input identified by the given layer name.
Implements ITfParser.
Definition at line 3694 of file TfParser.cpp.
|
overridevirtual |
Retrieves binding info (layer id and tensor info) for the network output identified by the given layer name.
Implements ITfParser.
Definition at line 3699 of file TfParser.cpp.
References CHECK_LOCATION, and IConnectableLayer::GetName().
|
friend |
Definition at line 96 of file TfParser.hpp.
Referenced by TfParser::TfParser().
|
friend |
Definition at line 97 of file TfParser.hpp.
Referenced by armnnTfParser::OutputShapeOfSqueeze().
|
friend |
Definition at line 98 of file TfParser.hpp.
Referenced by armnnTfParser::OutputShapeOfSqueeze().