Searched defs:batch_size_dim (Results 1 – 3 of 3) sorted by relevance
469 DimensionHandle batch_size_dim; in Conv2DShapeImpl() local606 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in Conv3DShape() local679 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in DepthwiseConv2DNativeShape() local759 DimensionHandle batch_size_dim = c->Dim( in AvgPoolShape() local911 DimensionHandle batch_size_dim = c->Dim( in MaxPoolShape() local1013 DimensionHandle batch_size_dim = c->Dim( in MaxPoolV2Shape() local1091 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in Pool3DShape() local
411 DimensionHandle batch_size_dim = c->Dim(padded, 0); in CommonFusedConvCalculations() local917 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in __anon3e672dd81d02() local
2468 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in __anon7c94107b3902() local2570 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); in __anon7c94107b3a02() local