Searched defs:in_cols_dim (Results 1 – 3 of 3) sorted by relevance
824 DimensionHandle in_cols_dim = c->Dim(input_shape, 3); in Conv3DShape() local1036 DimensionHandle in_cols_dim = c->Dim(input_shape, 2); in DepthwiseConv2DNativeShapeImpl() local1152 DimensionHandle in_cols_dim = c->Dim( in AvgPoolShape() local1631 DimensionHandle in_cols_dim = c->Dim( in MaxPoolShapeImpl() local1770 DimensionHandle in_cols_dim = c->Dim( in MaxPoolV2Shape() local1847 DimensionHandle in_cols_dim = c->Dim(input_shape, 3); in Pool3DShape() local
527 DimensionHandle in_cols_dim = c->Dim(padded, 2); in CommonFusedConvCalculations() local1047 DimensionHandle in_cols_dim = c->Dim(input_shape, 2); in __anon1bbab8941702() local
2662 DimensionHandle in_cols_dim = c->Dim(input_shape, 2); in __anon091a71b73902() local2765 DimensionHandle in_cols_dim = c->Dim(input_shape, 3); in __anon091a71b73a02() local