/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 87 b_rows, b_cols, p, a_indices.data(), a_values.data(), b.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...] |
/external/tensorflow/tensorflow/lite/kernels/internal/ |
tensor_utils_test.cc | 414 const int b_rows = 29, b_cols = 1, batches = 2; local 427 int8_t b_int8_data[b_rows * b_cols * batches]; 430 SymmetricQuantizeFloats(b_float_data, b_rows * b_cols, b_int8_data, &b_min, 432 SymmetricQuantizeFloats(&b_float_data[b_rows * b_cols], b_rows * b_cols, 433 &b_int8_data[b_rows * b_cols], &b_min, &b_max, 446 for (int i = 0; i < b_rows * b_cols * batches; ++i) { 453 float c_float_data[a_rows * b_cols * batches]; 454 for (int i = 0; i < a_rows * b_cols * batches; ++i) { 471 for (int i = 0; i < a_rows * b_cols * batches; ++i) [all...] |