Home
last modified time | relevance | path

Searched defs:rhs_ext1 (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dbatch_matmul.h72 const int rhs_ext1 = extent(extended_rhs_shape, 1); in BatchMatMul() local
163 const int rhs_ext1 = extent(extended_rhs_shape, 1); in BatchMatMul() local
317 const int rhs_ext1 = extent(extended_rhs_shape, 1); in BatchMatMul() local
423 const int rhs_ext1 = extent(extended_rhs_shape, 1); in BatchMatMul() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dbatch_matmul.h73 const int rhs_ext1 = batch_matmul::extent(extended_rhs_shape, 1); in BatchMatMul() local
131 const int rhs_ext1 = batch_matmul::extent(extended_rhs_shape, 1); in BatchMatMul() local
220 const int rhs_ext1 = batch_matmul::extent(extended_rhs_shape, 1); in BatchMatMul() local