Searched defs:in_cols_dim (Results 1 – 3 of 3) sorted by relevance
609 DimensionHandle in_cols_dim = c->Dim(input_shape, 3); in Conv3DShape() local681 DimensionHandle in_cols_dim = c->Dim(input_shape, 2); in DepthwiseConv2DNativeShape() local763 DimensionHandle in_cols_dim = c->Dim( in AvgPoolShape() local915 DimensionHandle in_cols_dim = c->Dim( in MaxPoolShape() local1017 DimensionHandle in_cols_dim = c->Dim( in MaxPoolV2Shape() local1094 DimensionHandle in_cols_dim = c->Dim(input_shape, 3); in Pool3DShape() local
413 DimensionHandle in_cols_dim = c->Dim(padded, 2); in CommonFusedConvCalculations() local919 DimensionHandle in_cols_dim = c->Dim(input_shape, 2); in __anon3e672dd81d02() local
2470 DimensionHandle in_cols_dim = c->Dim(input_shape, 2); in __anon7c94107b3902() local2573 DimensionHandle in_cols_dim = c->Dim(input_shape, 3); in __anon7c94107b3a02() local