Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dconv.h84 const int gemm_input_cols = FlatSizeSkipDim(*gemm_input_shape, 3); in ConvPerChannel() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1393 const int gemm_input_cols = filter_cols; in HybridConv() local
1475 const int gemm_input_cols = FlatSizeSkipDim(*gemm_input_shape, 3); in HybridConvPerChannel() local
1594 const int gemm_input_cols = gemm_input_shape->Dims(0) * in Conv() local
Dlegacy_optimized_ops.h2681 const int gemm_input_cols = gemm_input_shape->Dims(0) * in Conv() local