Searched defs:batch_size_dim (Results 1 – 3 of 3) sorted by relevance
636 DimensionHandle batch_size_dim; in Conv2DShapeImpl() local789 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in Conv3DShape() local867 DimensionHandle batch_size_dim; in Conv2DBackpropInputShape() local979 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in DepthwiseConv2DNativeShapeImpl() local1093 DimensionHandle batch_size_dim = c->Dim( in AvgPoolShape() local1536 DimensionHandle batch_size_dim = c->Dim( in MaxPoolShapeImpl() local1675 DimensionHandle batch_size_dim = c->Dim( in MaxPoolV2Shape() local1753 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in Pool3DShape() local
491 DimensionHandle batch_size_dim = c->Dim(padded, 0); in CommonFusedConvCalculations() local1011 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in __anon4650176f1702() local
2627 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in __anondb9326b23902() local2729 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in __anondb9326b23a02() local