Home
last modified time | relevance | path

Searched refs:b_cols (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dquantized_matmul_op_test.cc94 const int b_cols = 1; in TEST_F() local
120 AddInputFromArray<quint8>(TensorShape({b_rows, b_cols}), {0}); in TEST_F()
130 Tensor expected(allocator(), DT_QINT32, TensorShape({a_cols, b_cols})); in TEST_F()
144 const int b_cols = 1; in TEST_F() local
164 AddInputFromArray<quint8>(TensorShape({b_rows, b_cols}), {0}); in TEST_F()
184 const int b_cols = 2; in TEST_F() local
228 AddInputFromArray<quint8>(TensorShape({b_rows, b_cols}), { in TEST_F()
255 Tensor expected(allocator(), DT_QINT32, TensorShape({a_cols, b_cols})); in TEST_F()
308 const int b_cols = 8; in TEST_F() local
311 Tensor b_float(DT_FLOAT, {b_rows, b_cols}); in TEST_F()
[all …]
Dsparse_tensor_dense_matmul_op_gpu.cu.cc31 int nnz, int m, int b_rows, int b_cols, int p, in SparseTensorDenseMatMulKernel() argument
36 const int n = (ADJ_B) ? b_cols : b_rows; in SparseTensorDenseMatMulKernel()
56 const T b_value = ldg(b + ((ADJ_B) ? j * b_cols + k : k * b_cols + j)); in SparseTensorDenseMatMulKernel()
76 int b_cols = b.dimension(1); in Compute() local
85 b_rows, b_cols, p, a_indices.data(), a_values.data(), b.data(), in Compute()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtensor_utils_test.cc1402 const int b_rows = 29, b_cols = 1, batches = 2; in TEST() local
1415 int8_t b_int8_data[b_rows * b_cols * batches]; in TEST()
1418 SymmetricQuantizeFloats(b_float_data, b_rows * b_cols, b_int8_data, &b_min, in TEST()
1420 SymmetricQuantizeFloats(&b_float_data[b_rows * b_cols], b_rows * b_cols, in TEST()
1421 &b_int8_data[b_rows * b_cols], &b_min, &b_max, in TEST()
1434 for (int i = 0; i < b_rows * b_cols * batches; ++i) { in TEST()
1441 float c_float_data[a_rows * b_cols * batches]; in TEST()
1442 for (int i = 0; i < a_rows * b_cols * batches; ++i) { in TEST()
1458 for (int i = 0; i < a_rows * b_cols * batches; ++i) { in TEST()
1472 for (int i = 0; i < a_rows * b_cols * batches; ++i) { in TEST()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1541 Value b_cols = BroadcastConstant(loc, indices_shape, cols, 32, rewriter); in matchAndRewrite() local
1559 rewriter.create<SubOp>(loc, b_cols, in matchAndRewrite()
1601 rewriter.create<TF::LessOp>(loc, b_false.getType(), n_plus_x, b_cols)); in matchAndRewrite()