Home
last modified time | relevance | path

Searched refs:num_full_planes (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dconvert.cc246 const int num_full_planes = shape.c / kPhwc4ChannelsInPlane; in ConvertToPHWC4() local
250 for (int p = 0; p < num_full_planes; p++) { in ConvertToPHWC4()
265 shape.c - num_full_planes * kPhwc4ChannelsInPlane; in ConvertToPHWC4()
272 shape.LinearIndex({b, 0, 0, num_full_planes * kPhwc4ChannelsInPlane}); in ConvertToPHWC4()
274 num_pixels * num_full_planes * kPhwc4ChannelsInPlane; in ConvertToPHWC4()
295 const int num_full_planes = shape.c / kPhwc4ChannelsInPlane; in ConvertToPHWC4Half() local
299 for (int p = 0; p < num_full_planes; p++) { in ConvertToPHWC4Half()
317 shape.c - num_full_planes * kPhwc4ChannelsInPlane; in ConvertToPHWC4Half()
325 shape.LinearIndex({b, 0, 0, num_full_planes * kPhwc4ChannelsInPlane}); in ConvertToPHWC4Half()
327 num_pixels * num_full_planes * kPhwc4ChannelsInPlane; in ConvertToPHWC4Half()
[all …]