D | BNNMTest.java | 129 private byte[] runBNNM(int m, int n, int k, byte[] a_byte, int a_offset, byte[] b_byte, in runBNNM() argument 133 Type a_type = builder.setX(k).setY(m).create(); in runBNNM() 134 Type b_type = builder.setX(k).setY(n).create(); in runBNNM() 206 final int k = a_rows; in testSmallMatrices() local 208 byte[] c_byte_output = runBNNM(m, n, k, a_data, a_offset, b_data, b_offset, in testSmallMatrices() 251 final int k = a_rows; in testMediumMatrices1() local 253 byte[] c_byte_output = runBNNM(m, n, k, a_data, a_offset, b_data, b_offset, in testMediumMatrices1() 303 final int k = a_rows; in testMediumMatrices2() local 305 byte[] c_byte_output = runBNNM(m, n, k, a_data, a_offset, b_data, b_offset, in testMediumMatrices2() 319 int k = 1152; in testRealData() local [all …]
|