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

Function BuildOpInfoWithoutDevice

tensorflow/core/grappler/costs/utils.cc:269–281  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

267}
268
269OpInfo BuildOpInfoWithoutDevice(
270 const NodeDef& node,
271 const std::unordered_map<string, const NodeDef*>& name_to_node,
272 const std::vector<OpInfo::TensorProperties>& inputs) {
273 OpInfo op_info;
274 op_info.set_op(node.op());
275 *op_info.mutable_attr() = node.attr();
276 for (auto& input : inputs) {
277 *op_info.add_inputs() = input;
278 }
279 ExtractExtraProperties(node, name_to_node, &op_info);
280 return op_info;
281}
282
283string GetOpDescription(const OpInfo& op_info) {
284 string description = "[";

Callers 2

TESTFunction · 0.85

Calls 5

ExtractExtraPropertiesFunction · 0.85
set_opMethod · 0.80
attrMethod · 0.80
opMethod · 0.45
add_inputsMethod · 0.45

Tested by 1

TESTFunction · 0.68