Function TensorLayoutInfo getTensorLayoutInfo

core/tensor_layout.h:390–421

Returns information about the tensor layout

  // Returns information about the tensor layout
  oidn_inline TensorLayoutInfo getTensorLayoutInfo(TensorLayout layout)
  {
    switch (layout)
    {
    case TensorLayout::x:
      return {1, 1};
    case TensorLayout::chw:
    case TensorLayout::hwc:
      return {3, 1};
    case TensorLayout::Chw8c:
      return {3, 8};
    case TensorLayout::Chw16c:
      return {3, 16};
    case TensorLayout::Chw32c:
      return {3, 32};
    case TensorLayout::oihw:
    case TensorLayout::ohwi:
      return {4, 1};
    case TensorLayout::IOhw8i8o:
    case TensorLayout::OIhw8i8o:
      return {4, 8};
    case TensorLayout::IOhw16i16o:
    case TensorLayout::OIhw16i16o:
    case TensorLayout::OIhw2o8i8o2i:
    case TensorLayout::OIhw8i16o2i:
      return {4, 16};
    case TensorLayout::OIhw2o16i16o2i:
      return {4, 32};
    default:
      throw std::invalid_argument("invalid tensor layout");
    }
  }

  inline std::ostream& operator <<(std::ostream& sm, TensorLayout layout)
  {