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

Function GetArithmeticOpsCount

tensorflow/lite/toco/dump_graphviz.cc:665–677  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

663}
664
665int64 GetArithmeticOpsCount(const Model& model, const string& array_id) {
666 for (const auto& op : model.operators) {
667 if (!op->outputs.empty() && op->outputs[0] == array_id) {
668 int64 count;
669 if (EstimateArithmeticOpsCount(model, *op, &count)) {
670 return count;
671 } else {
672 return 0;
673 }
674 }
675 }
676 return 0;
677}
678
679void InsertNode(const Model& model, const string& array_id, Node* node,
680 std::vector<string> prefixes, int64* math_ops) {

Callers 1

InsertNodeFunction · 0.85

Calls 2

emptyMethod · 0.45

Tested by

no test coverage detected