code: def _update_input_size(input_size, stride):
      sh, sw = (stride, stride) if isinstance(stride, int) else stride
      ih, iw = (
          (input_size, input_size)
          if isinstance(input_size, int)
          else input_size
      )
      oh, ow = math.ceil(ih / sh), math.ceil(iw / sw)
      input_size = (oh, ow)
      return input_size
  
imports: from __future__ import division
  import math
  from mxnet.context import cpu
  from mxnet.gluon import nn
  from mxnet.gluon.block import HybridBlock
  from mxnet.gluon.nn import BatchNorm
  from ..nn.dropblock import DropBlock
  from ..nn.splat import SplitAttentionConv
label_desc: Use image node_id 1 for calling a global function with example usage: _update_input_size(input_size, stride) and returns: input_size