| 2191 | } |
| 2192 | |
| 2193 | int AxesCount(AxesOrder axes_order) { |
| 2194 | switch (axes_order) { |
| 2195 | case AxesOrder::kOneAxis: |
| 2196 | return 1; |
| 2197 | case AxesOrder::kRC: |
| 2198 | return 2; |
| 2199 | case AxesOrder::kCR: |
| 2200 | return 2; |
| 2201 | case AxesOrder::kHWIO: |
| 2202 | return 4; |
| 2203 | case AxesOrder::kOHWI: |
| 2204 | return 4; |
| 2205 | case AxesOrder::kHWIM: |
| 2206 | return 4; |
| 2207 | case AxesOrder::k1HWO: |
| 2208 | return 4; |
| 2209 | case AxesOrder::kNHWC: |
| 2210 | return 4; |
| 2211 | case AxesOrder::kHWOI: |
| 2212 | return 4; |
| 2213 | default: |
| 2214 | LOG(FATAL) << "Bad AxesOrder"; |
| 2215 | return 0; |
| 2216 | } |
| 2217 | } |
| 2218 | |
| 2219 | bool IsDiscardableArray(const Model& model, const string& array_name) { |
| 2220 | if (IsInputArray(model, array_name) || IsOutputArray(model, array_name)) { |
no outgoing calls
no test coverage detected