Searched defs:output_image_size (Results 1 – 13 of 13) sorted by relevance
55 int64_t output_image_size = output.numel() / channels; in cpu_max_unpool() local119 int64_t output_image_size = output_height * output_width; in cpu_max_unpool_channels_last() local192 int64_t output_image_size = grad_output.numel() / channels; in cpu_max_unpool_backward() local
427 const int64_t output_image_size = in operator ()() local805 const int64_t output_image_size = in ComputeBackpropFilter() local916 const int64_t output_image_size = in operator ()() local
247 const int32_t output_image_size = in SpatialMaxPool() local
200 const int64_t output_image_size = in operator ()() local
127 const int32_t output_image_size = out_height * out_width * depth; in SpatialMaxPoolWithArgMaxHelper() local629 const int32_t output_image_size = out_height * out_width * params.depth; in SpatialMaxPoolGradGrad() local
439 const int64_t output_image_size = dims.spatial_dims[0].output_size * in Compute() local963 const int64_t output_image_size = dims.spatial_dims[0].output_size * in Compute() local
616 const int32_t output_image_size = in launch() local
511 const int output_image_size = in Compute() local
1133 const int64_t output_image_size = in operator ()() local
116 let output_image_size = output_format.plane_fmt[0].sizeimage as usize; in run() localVariable
161 let output_image_size = output_format.plane_fmt[0].sizeimage as usize; in run() localVariable
47 const int output_image_size = output_height * output_width; in TransposeConvV2() local
4968 const int output_image_size = output_height * output_width; in TransposeConvV2() local5467 const int output_image_size = output_height * output_width; in TransposeConvV2() local