Searched defs:b_rows (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantized_matmul_op_test.cc | 93 const int b_rows = 1; in TEST_F() local
143 const int b_rows = 1; in TEST_F() local
182 const int b_rows = 1; in TEST_F() local
221 const int b_rows = 3; in TEST_F() local
345 const int b_rows = 8; in TEST_F() local
|
| H A D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 47 int nnz, int m, int b_rows, int b_cols, int p, in SparseTensorDenseMatMulKernel()
93 int b_rows = b.dimension(0); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | tensor_utils_test.cc | 1398 const int b_rows = 29, b_cols = 1, batches = 2; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 1941 Value b_rows = BroadcastConstant(loc, indices_shape, rows, 32, rewriter); in matchAndRewrite() local
|