Searched defs:in_cols_dim (Results 1 – 3 of 3) sorted by relevance
792 DimensionHandle in_cols_dim = c->Dim(input_shape, 3); in Conv3DShape() local981 DimensionHandle in_cols_dim = c->Dim(input_shape, 2); in DepthwiseConv2DNativeShapeImpl() local1097 DimensionHandle in_cols_dim = c->Dim( in AvgPoolShape() local1540 DimensionHandle in_cols_dim = c->Dim( in MaxPoolShapeImpl() local1679 DimensionHandle in_cols_dim = c->Dim( in MaxPoolV2Shape() local1756 DimensionHandle in_cols_dim = c->Dim(input_shape, 3); in Pool3DShape() local
493 DimensionHandle in_cols_dim = c->Dim(padded, 2); in CommonFusedConvCalculations() local1013 DimensionHandle in_cols_dim = c->Dim(input_shape, 2); in __anonad10f3361702() local
2629 DimensionHandle in_cols_dim = c->Dim(input_shape, 2); in __anon42d741193902() local2732 DimensionHandle in_cols_dim = c->Dim(input_shape, 3); in __anon42d741193a02() local