Searched refs:gemm_input_shape (Results 1 – 3 of 3) sorted by relevance
54 const RuntimeShape* gemm_input_shape = nullptr; in ConvPerChannel() local70 gemm_input_shape = &im2col_shape; in ConvPerChannel()76 gemm_input_shape = &im2col_shape; in ConvPerChannel()80 gemm_input_shape = &input_shape; in ConvPerChannel()83 const int gemm_input_rows = gemm_input_shape->Dims(3); in ConvPerChannel()84 const int gemm_input_cols = FlatSizeSkipDim(*gemm_input_shape, 3); in ConvPerChannel()
2445 const RuntimeShape* gemm_input_shape = nullptr; in Conv() local2456 gemm_input_shape = &im2col_shape; in Conv()2462 gemm_input_shape = &im2col_shape; in Conv()2468 gemm_input_shape = &input_shape; in Conv()2480 const int gemm_input_dims = gemm_input_shape->DimensionsCount(); in Conv()2481 int m = FlatSizeSkipDim(*gemm_input_shape, gemm_input_dims - 1); in Conv()2483 int k = gemm_input_shape->Dims(gemm_input_dims - 1); in Conv()2646 const RuntimeShape* gemm_input_shape = nullptr; in Conv() local2661 gemm_input_shape = &im2col_shape; in Conv()2670 gemm_input_shape = &im2col_shape; in Conv()[all …]
1259 const RuntimeShape* gemm_input_shape = nullptr; in Conv() local1270 gemm_input_shape = &im2col_shape; in Conv()1276 gemm_input_shape = &im2col_shape; in Conv()1280 gemm_input_shape = &input_shape; in Conv()1283 const int gemm_input_dims = gemm_input_shape->DimensionsCount(); in Conv()1284 int m = FlatSizeSkipDim(*gemm_input_shape, gemm_input_dims - 1); in Conv()1286 int k = gemm_input_shape->Dims(gemm_input_dims - 1); in Conv()1445 const RuntimeShape* gemm_input_shape = nullptr; in HybridConvPerChannel() local1461 gemm_input_shape = &im2col_shape; in HybridConvPerChannel()1466 gemm_input_shape = &im2col_shape; in HybridConvPerChannel()[all …]