| 103 | } |
| 104 | |
| 105 | void initData(size_t M, size_t N, std::unique_ptr<float[]> &inputPtr) { |
| 106 | std::mt19937 gen(314159); |
| 107 | randn(inputPtr.get(), M * N, gen); |
| 108 | LOG(kDefLog, kInfo, "%s", show<float>(inputPtr.get(), M, N, "Input").c_str()); |
| 109 | } |
| 110 | |
| 111 | Kernel selectTranspose(Context &ctx, int version, |
| 112 | const Bindings</* input, output */ 2> &bindings, |