| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_tensor_dense_matmul_op_gpu.cu.cc | 31 __global__ void SparseTensorDenseMatMulKernel(int nnz, int m, int b_rows,
38 const int n = (ADJ_B) ? b_cols : b_rows;
77 int b_rows = b.dimension(0); local
87 b_rows, b_cols, p, a_indices.data(), a_values.data(), b.data(),
|
| quantized_matmul_op_test.cc | 93 const int b_rows = 1; local
120 AddInputFromArray<quint8>(TensorShape({b_rows, b_cols}), {0});
143 const int b_rows = 1; local
164 AddInputFromArray<quint8>(TensorShape({b_rows, b_cols}), {0});
183 const int b_rows = 3; local
228 AddInputFromArray<quint8>(TensorShape({b_rows, b_cols}), {
307 const int b_rows = 8; local
311 Tensor b_float(DT_FLOAT, {b_rows, b_cols});
|
| /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) {
[all...] |
