MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GetNodeAttr

Function GetNodeAttr

serving/processor/framework/graph_optimizer.cc:877–888  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

875}
876
877Status GetNodeAttr(const Node* node,
878 const std::string& attr_name,
879 AttrValue** attr) {
880 *attr = const_cast<AttrValue*>(node->attrs().Find(attr_name));
881 if (*attr == nullptr) {
882 return tensorflow::errors::Internal(
883 "Miss attr: ", attr_name ," in the node, ",
884 node->DebugString());
885 }
886
887 return Status::OK();
888}
889
890 Status AddPlaceholder(Graph* g, const std::string& name,
891 DataType type, Node** target_node) {

Callers 7

ModifySharedNameAttrFunction · 0.70
ConvertToHashLookupOpMethod · 0.70
ConvertToHashImportOpMethod · 0.70
CreateIncrRestoreOpMethod · 0.70

Calls 4

InternalFunction · 0.85
FindMethod · 0.45
attrsMethod · 0.45
DebugStringMethod · 0.45

Tested by

no test coverage detected