| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_tensor_dense_matmul_op_gpu.cu.cc | 32 int b_cols, int p,
38 const int n = (ADJ_B) ? b_cols : b_rows;
58 const T b_value = ldg(b + ((ADJ_B) ? j * b_cols + k : k * b_cols + j));
78 int b_cols = b.dimension(1); local
86 nnz, m, b_rows, b_cols, p, a_indices.data(), a_values.data(),
|
| quantized_matmul_op_test.cc | 94 const int b_cols = 1; local
120 AddInputFromArray<quint8>(TensorShape({b_rows, b_cols}), {0});
130 Tensor expected(allocator(), DT_QINT32, TensorShape({a_cols, b_cols}));
144 const int b_cols = 1; local
164 AddInputFromArray<quint8>(TensorShape({b_rows, b_cols}), {0});
184 const int b_cols = 2; local
228 AddInputFromArray<quint8>(TensorShape({b_rows, b_cols}), {
255 Tensor expected(allocator(), DT_QINT32, TensorShape({a_cols, b_cols}));
308 const int b_cols = 8; local
311 Tensor b_float(DT_FLOAT, {b_rows, b_cols});
[all...] |
| /cts/tests/tests/rsblas/src/android/renderscript/cts/ |
| BNNMTest.java | 180 final int b_cols = 4; local
205 final int n = b_cols;
236 final int b_cols = 7; local
250 final int n = b_cols;
284 final int b_cols = 9; local
302 final int n = b_cols;
367 final int b_cols = 4; local
392 final int n = b_cols;
422 final int b_cols = 4; local
446 final int n = b_cols;
[all...] |
| /frameworks/rs/tests/java_api/RsBLAS_Benchmark/src/com/example/android/rs/blasbenchmark/ |
| BNNMTest.java | 194 final int b_cols = 4; local
219 n = b_cols;
271 final int b_cols = 9; local
289 n = b_cols;
|
