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

Function CopyNodeAttr

tensorflow/tools/graph_transforms/transform_utils.cc:146–151  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

144}
145
146void CopyNodeAttr(const NodeDef& source, const string& source_key,
147 const string& dest_key, NodeDef* dest) {
148 CHECK_NE(0, source.attr().count(source_key))
149 << "No key '" << source_key << "' found in " << source.DebugString();
150 (*(dest->mutable_attr()))[dest_key] = source.attr().at(source_key);
151}
152
153Tensor GetNodeTensorAttr(const NodeDef& node, const string& key) {
154 TensorProto tensor_proto = node.attr().at(key).tensor();

Callers 7

QuantizeNodesFunction · 0.85
FlattenAtrousConvFunction · 0.85
TestCopyNodeAttrMethod · 0.85
FuseResizePadAndConvFunction · 0.85
FuseResizeAndConvFunction · 0.85
FusePadAndConvFunction · 0.85

Calls 4

attrMethod · 0.80
countMethod · 0.45
DebugStringMethod · 0.45
atMethod · 0.45

Tested by 1

TestCopyNodeAttrMethod · 0.68