Searched refs:kPhwc4ChannelsInPlane (Results 1 – 1 of 1) sorted by relevance
38 constexpr int kPhwc4ChannelsInPlane = 4; variable243 int num_planes = DivideRoundUp(shape.c, kPhwc4ChannelsInPlane); in ConvertToPHWC4()246 const int num_full_planes = shape.c / kPhwc4ChannelsInPlane; in ConvertToPHWC4()249 out.data() + b * num_pixels * num_planes * kPhwc4ChannelsInPlane; in ConvertToPHWC4()252 in.data() + shape.LinearIndex({b, 0, 0, p * kPhwc4ChannelsInPlane}); in ConvertToPHWC4()254 std::memcpy(dest, src, kPhwc4ChannelsInPlane * sizeof(float)); in ConvertToPHWC4()256 dest += kPhwc4ChannelsInPlane; in ConvertToPHWC4()263 const int padded_size = num_pixels * num_planes * kPhwc4ChannelsInPlane; in ConvertToPHWC4()265 shape.c - num_full_planes * kPhwc4ChannelsInPlane; in ConvertToPHWC4()272 shape.LinearIndex({b, 0, 0, num_full_planes * kPhwc4ChannelsInPlane}); in ConvertToPHWC4()[all …]