| 105 | } |
| 106 | |
| 107 | uint3 Get3DSizeForPHWO4I4(const OHWI& shape) { |
| 108 | return uint3(AlignByN(shape.i, 4), shape.h * shape.w, |
| 109 | IntegralDivideRoundUp(shape.o, 4)); |
| 110 | } |
| 111 | |
| 112 | // Layout is Po,H,W,OI4x4. |
| 113 | Status ConvertToPHWO4I4(absl::Span<const float> in, const IHWO& shape, |
no test coverage detected