Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dbatch_matmul.h73 const int rhs_ext2 = extent(extended_rhs_shape, 2); in BatchMatMul() local
164 const int rhs_ext2 = extent(extended_rhs_shape, 2); in BatchMatMul() local
318 const int rhs_ext2 = extent(extended_rhs_shape, 2); in BatchMatMul() local
424 const int rhs_ext2 = extent(extended_rhs_shape, 2); in BatchMatMul() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dbatch_matmul.h74 const int rhs_ext2 = batch_matmul::extent(extended_rhs_shape, 2); in BatchMatMul() local
132 const int rhs_ext2 = batch_matmul::extent(extended_rhs_shape, 2); in BatchMatMul() local
221 const int rhs_ext2 = batch_matmul::extent(extended_rhs_shape, 2); in BatchMatMul() local