Home
last modified time | relevance | path

Searched defs:b_cols (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantized_matmul_op_test.cc94 const int b_cols = 1; in TEST_F() local
144 const int b_cols = 1; in TEST_F() local
183 const int b_cols = 1; in TEST_F() local
222 const int b_cols = 2; in TEST_F() local
346 const int b_cols = 8; in TEST_F() local
H A Dsparse_tensor_dense_matmul_op_gpu.cu.cc47 int nnz, int m, int b_rows, int b_cols, int p, in SparseTensorDenseMatMulKernel()
94 int b_cols = b.dimension(1); in Compute() local
/aosp_15_r20/cts/tests/tests/rsblas/src/android/renderscript/cts/
H A DBNNMTest.java180 final int b_cols = 4; in testSmallMatrices() local
236 final int b_cols = 7; in testMediumMatrices1() local
284 final int b_cols = 9; in testMediumMatrices2() local
367 final int b_cols = 4; in testClamping() local
422 final int b_cols = 4; in testExceptionHandling() local
/aosp_15_r20/frameworks/rs/tests/java_api/RsBLAS_Benchmark/src/com/example/android/rs/blasbenchmark/
H A DBNNMTest.java194 final int b_cols = 4; in setTestSmall() local
271 final int b_cols = 9; in setTestMedium() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dtensor_utils_test.cc1398 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 Dlegalize_tf.cc1942 Value b_cols = BroadcastConstant(loc, indices_shape, cols, 32, rewriter); in matchAndRewrite() local