Searched defs:output_image_size (Results 1 – 9 of 9) sorted by relevance
211 const int32 output_image_size = out_height * out_width * params.depth; in SpatialMaxPool() local
411 const int64 output_image_size = in operator ()() local760 const int64 output_image_size = in ComputeBackpropFilter() local871 const int64 output_image_size = in operator ()() local
192 const int64 output_image_size = in operator ()() local
117 const int32 output_image_size = out_height * out_width * depth; in SpatialMaxPoolWithArgMaxHelper() local597 const int32 output_image_size = out_height * out_width * params.depth; in SpatialMaxPoolGradGrad() local
380 const int64 output_image_size = dims.spatial_dims[0].output_size * in Compute() local847 const int64 output_image_size = dims.spatial_dims[0].output_size * in Compute() local
597 const int32 output_image_size = in launch() local
300 const int output_image_size = in Compute() local
396 const int output_image_size = in Compute() local
1115 const int64 output_image_size = args.out_rows * args.out_cols * out_depth; in operator ()() local