Searched refs:kPhwo4i4ChannelsInPlane (Results 1 – 1 of 1) sorted by relevance
39 constexpr int kPhwo4i4ChannelsInPlane = 4; variable57 for (int p = 0; p < DivideRoundUp(shape.o, kPhwo4i4ChannelsInPlane); ++p) { in ConvertToPHWO4I4()60 for (int c = 0; c < DivideRoundUp(shape.i, kPhwo4i4ChannelsInPlane); in ConvertToPHWO4I4()62 for (int co = 0; co < kPhwo4i4ChannelsInPlane; ++co) { in ConvertToPHWO4I4()63 for (int ci = 0; ci < kPhwo4i4ChannelsInPlane; ++ci) { in ConvertToPHWO4I4()65 if (c * kPhwo4i4ChannelsInPlane + ci < shape.i && in ConvertToPHWO4I4()66 p * kPhwo4i4ChannelsInPlane + co < shape.o) { in ConvertToPHWO4I4()68 int tensor_o = p * kPhwo4i4ChannelsInPlane + co; in ConvertToPHWO4I4()69 int tensor_i = c * kPhwo4i4ChannelsInPlane + ci; in ConvertToPHWO4I4()87 return AlignByN(shape.i, kPhwo4i4ChannelsInPlane) * in GetElementsSizeForPHWO4I4()[all …]