onert::shape_inference::bcq Namespace Reference

Functions
int	getOutputSize (const ir::Shape &cluster_shape, const int32_t *cluster_buf)

Function Documentation

getOutputSize()

int onert::shape_inference::bcq::getOutputSize ( const ir::Shape &  cluster_shape, const int32_t *  cluster_buf  )

inline

Definition at line 77 of file ShapeInference.cc.

{
  int size = 0;
  for (int idx = 0; idx < cluster_shape.dim(0); idx++)
  {
    size += cluster_buf[idx * 2 + 1];
  }
  return size;
}

References size.

Referenced by onert::shape_inference::inferBCQFullyConnectedShape(), and onert::shape_inference::inferBCQGatherShape().