Home
last modified time | relevance | path

Searched defs:gemm_input_rows (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A Dconv.h84 const int gemm_input_rows = gemm_input_shape->Dims(3); in ConvPerChannel() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Doptimized_ops.h1065 const int gemm_input_rows = num_input / gemm_input_cols; in HybridConv() local
1145 const int gemm_input_rows = gemm_input_shape->Dims(3); in HybridConvPerChannel() local
1260 const int gemm_input_rows = gemm_input_shape->Dims(3); in Conv() local
H A Dlegacy_optimized_ops.h2678 const int gemm_input_rows = gemm_input_shape->Dims(3); in Conv() local