Searched refs:gemm_input_cols (Results 1 – 3 of 3) sorted by relevance
84 const int gemm_input_cols = FlatSizeSkipDim(*gemm_input_shape, 3); in ConvPerChannel() local93 TFLITE_DCHECK_EQ(output_cols, gemm_input_cols); in ConvPerChannel()104 rhs_params.cols = gemm_input_cols; in ConvPerChannel()
1395 const int gemm_input_cols = filter_cols; in HybridConv() local1396 const int gemm_input_rows = num_input / gemm_input_cols; in HybridConv()1477 const int gemm_input_cols = FlatSizeSkipDim(*gemm_input_shape, 3); in HybridConvPerChannel() local1483 TFLITE_DCHECK_EQ(output_cols, gemm_input_cols); in HybridConvPerChannel()1503 rhs_params.cols = gemm_input_cols; in HybridConvPerChannel()1596 const int gemm_input_cols = gemm_input_shape->Dims(0) * in Conv() local1610 TFLITE_DCHECK_EQ(output_cols, gemm_input_cols); in Conv()1621 rhs_params.cols = gemm_input_cols; in Conv()
2682 const int gemm_input_cols = gemm_input_shape->Dims(0) * in Conv() local2696 TFLITE_DCHECK_EQ(output_cols, gemm_input_cols); in Conv()2701 if (gemm_input_cols == 1 && output_rows >= 4) { in Conv()2717 gemm_input_data, gemm_input_rows, gemm_input_cols); in Conv()