Searched defs:batch_size_dim (Results 1 – 3 of 3) sorted by relevance
644 DimensionHandle batch_size_dim; in Conv2DShapeImpl() local821 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in Conv3DShape() local903 DimensionHandle batch_size_dim; in Conv2DBackpropInputShape() local1034 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in DepthwiseConv2DNativeShapeImpl() local1148 DimensionHandle batch_size_dim = c->Dim( in AvgPoolShape() local1627 DimensionHandle batch_size_dim = c->Dim( in MaxPoolShapeImpl() local1766 DimensionHandle batch_size_dim = c->Dim( in MaxPoolV2Shape() local1844 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in Pool3DShape() local
525 DimensionHandle batch_size_dim = c->Dim(padded, 0); in CommonFusedConvCalculations() local1045 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in __anon1bbab8941702() local
2660 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in __anon091a71b73902() local2762 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in __anon091a71b73a02() local