Searched refs:nodeDef (Results 1 – 2 of 2) sorted by relevance
| /external/armnn/src/armnnTfParser/ |
| D | TfParser.hpp | 109 void LoadNodeDef(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef);
112 const tensorflow::NodeDef* ResolveIdentityNode(const tensorflow::NodeDef* nodeDef);
114 std::vector<OutputOfConstNodeDef> GetTfInputNodes(const tensorflow::NodeDef& nodeDef) const;
119 …or<OutputOfParsedTfOperation> GetInputParsedTfOperationsChecked(const tensorflow::NodeDef& nodeDef,
122 …ParsedTfOperationPtr ParseConst(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& gr…
132 …ParsedTfOperationPtr ParseAdd(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& grap…
133 …ParsedTfOperationPtr ParseAddN(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& gra…
134 …ParsedTfOperationPtr ParseBiasAdd(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& …
135 …ParsedTfOperationPtr ParseConv2D(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& g…
136 ParsedTfOperationPtr ParseDepthwiseConv2D(const tensorflow::NodeDef& nodeDef,
[all …]
|
| D | TfParser.cpp | 43 void ReadMandatoryNodeAttributeImpl(const tensorflow::NodeDef& nodeDef, in ReadMandatoryNodeAttributeImpl() argument
48 auto iter = nodeDef.attr().find(attribName); in ReadMandatoryNodeAttributeImpl()
49 if (iter != nodeDef.attr().end()) in ReadMandatoryNodeAttributeImpl()
62 nodeDef.name(), in ReadMandatoryNodeAttributeImpl()
73 nodeDef.name(), in ReadMandatoryNodeAttributeImpl()
79 void ReadOptionalNodeAttributeImpl(const tensorflow::NodeDef& nodeDef, in ReadOptionalNodeAttributeImpl() argument
84 auto iter = nodeDef.attr().find(attribName); in ReadOptionalNodeAttributeImpl()
85 if (iter != nodeDef.attr().end()) in ReadOptionalNodeAttributeImpl()
98 nodeDef.name(), in ReadOptionalNodeAttributeImpl()
106 float ReadMandatoryNodeFloatAttribute(const tensorflow::NodeDef& nodeDef, const std::string& name) in ReadMandatoryNodeFloatAttribute() argument
[all …]
|