| 70 | }; |
| 71 | |
| 72 | inline DataLayout StringToDataLayout(const std::string& str) { |
| 73 | std::string s(str); |
| 74 | for (size_t i = 0; i < s.size(); ++i) { |
| 75 | s[i] = toupper(s[i]); |
| 76 | } |
| 77 | |
| 78 | if (s == "NHWC") { |
| 79 | return DataLayout::NHWC; |
| 80 | } else if (s == "NCHW") { |
| 81 | return DataLayout::NCHW; |
| 82 | } else if (s == "ANYLAYOUT") { |
| 83 | return DataLayout::ANY; |
| 84 | } else if (s == "UNDEFINED(ANYLAYOUT)") { |
| 85 | return DataLayout::ANY; |
| 86 | } else if (s == "MKLDNNLAYOUT") { |
| 87 | return DataLayout::ONEDNN; |
| 88 | } else if (s == "ONEDNNLAYOUT") { |
| 89 | return DataLayout::ONEDNN; |
| 90 | } else if (s == "SPARSE_COO") { |
| 91 | return DataLayout::SPARSE_COO; |
| 92 | } else if (s == "SPARSE_CSR") { |
| 93 | return DataLayout::SPARSE_CSR; |
| 94 | } else if (s == "NDHWC") { |
| 95 | return DataLayout::kNDHWC; |
| 96 | } else if (s == "PSTRING_UNION") { |
| 97 | return DataLayout::PSTRING_UNION; |
| 98 | } else if (s == "NCDHW") { |
| 99 | return DataLayout::NCDHW; |
| 100 | } else if (s == "STRIDED") { |
| 101 | return DataLayout::STRIDED; |
| 102 | } else { |
| 103 | PD_THROW("Unknown data layout type string: ", s, "."); |
| 104 | } |
| 105 | } |
| 106 | |
| 107 | inline std::string DataLayoutToString(const DataLayout& layout) { |
| 108 | switch (layout) { |
no test coverage detected