| 1146 | } |
| 1147 | |
| 1148 | StatusOr<bool> IndexedArrayAnalysisPrinterPass::Run(HloModule* module) { |
| 1149 | if (!VLOG_IS_ON(2)) { |
| 1150 | return false; |
| 1151 | } |
| 1152 | |
| 1153 | IndexedArrayAnalysis analysis; |
| 1154 | for (auto* computation : module->MakeNonfusionComputations()) { |
| 1155 | for (auto* instr : computation->instructions()) { |
| 1156 | TF_ASSIGN_OR_RETURN(Analysis::Array * t, analysis.GetArrayFor(instr)); |
| 1157 | if (!dynamic_cast<UnknownArray*>(t) && !dynamic_cast<ConstantArray*>(t)) { |
| 1158 | VLOG(2) << instr->ToString() << " -> " << analysis.ToString(t); |
| 1159 | } |
| 1160 | } |
| 1161 | } |
| 1162 | |
| 1163 | return false; |
| 1164 | } |
| 1165 | |
| 1166 | } // namespace xla |
nothing calls this directly
no test coverage detected