1 op { 2 graph_op_name: "AvgPool" 3 in_arg { 4 name: "value" 5 description: <<END 6 4-D with shape `[batch, height, width, channels]`. 7 END 8 } 9 out_arg { 10 name: "output" 11 description: <<END 12 The average pooled output tensor. 13 END 14 } 15 attr { 16 name: "ksize" 17 description: <<END 18 The size of the sliding window for each dimension of `value`. 19 END 20 } 21 attr { 22 name: "strides" 23 description: <<END 24 The stride of the sliding window for each dimension of `value`. 25 END 26 } 27 attr { 28 name: "padding" 29 description: <<END 30 The type of padding algorithm to use. 31 END 32 } 33 attr { 34 name: "data_format" 35 description: <<END 36 Specify the data format of the input and output data. With the 37 default format "NHWC", the data is stored in the order of: 38 [batch, in_height, in_width, in_channels]. 39 Alternatively, the format could be "NCHW", the data storage order of: 40 [batch, in_channels, in_height, in_width]. 41 END 42 } 43 summary: "Performs average pooling on the input." 44 description: <<END 45 Each entry in `output` is the mean of the corresponding size `ksize` 46 window in `value`. 47 END 48 } 49