| 38 | } |
| 39 | |
| 40 | string ToString(TensorFormat format) { |
| 41 | switch (format) { |
| 42 | case FORMAT_NHWC: |
| 43 | return "NHWC"; |
| 44 | case FORMAT_NCHW: |
| 45 | return "NCHW"; |
| 46 | case FORMAT_NCHW_VECT_C: |
| 47 | return "NCHW_VECT_C"; |
| 48 | case FORMAT_NHWC_VECT_W: |
| 49 | return "NHWC_VECT_W"; |
| 50 | case FORMAT_HWNC: |
| 51 | return "HWNC"; |
| 52 | case FORMAT_HWCN: |
| 53 | return "HWCN"; |
| 54 | default: |
| 55 | LOG(FATAL) << "Invalid Format: " << static_cast<int32>(format); |
| 56 | return "INVALID_FORMAT"; |
| 57 | } |
| 58 | } |
| 59 | |
| 60 | string ToString(FilterTensorFormat format) { |
| 61 | switch (format) { |
no outgoing calls