| 27 | ModelAnalyzer::ModelAnalyzer(const GrapplerItem& item) : item_(item) {} |
| 28 | |
| 29 | Status ModelAnalyzer::GenerateReport(bool debug, bool assume_valid_feeds, |
| 30 | std::ostream& os) { |
| 31 | GraphProperties properties(item_); |
| 32 | TF_RETURN_IF_ERROR(properties.InferStatically(assume_valid_feeds)); |
| 33 | |
| 34 | for (const auto& node : item_.MainOpsFanin()) { |
| 35 | PrintNodeInfo(node, properties, debug, os); |
| 36 | } |
| 37 | for (const auto& node : item_.EnqueueOpsFanin()) { |
| 38 | PrintNodeInfo(node, properties, debug, os); |
| 39 | } |
| 40 | |
| 41 | return Status::OK(); |
| 42 | } |
| 43 | |
| 44 | void ModelAnalyzer::PrintNodeInfo(const NodeDef* node, |
| 45 | const GraphProperties& properties, bool debug, |
nothing calls this directly
no test coverage detected