Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dconvert.cc50 if (out.size() != GetElementsSizeForPHWO4I4(shape)) { in ConvertToPHWO4I4()
53 out.size(), " != ", GetElementsSizeForPHWO4I4(shape))); in ConvertToPHWO4I4()
86 uint32_t GetElementsSizeForPHWO4I4(const OHWI& shape) { in GetElementsSizeForPHWO4I4() function
91 uint32_t GetElementsSizeForPHWO4I4(const IHWO& shape) { in GetElementsSizeForPHWO4I4() function
98 std::vector<float> transposed(GetElementsSizeForPHWO4I4(tensor.shape)); in ConvertToPHWO4I4()
108 std::vector<float> transposed(GetElementsSizeForPHWO4I4(tensor.shape)); in ConvertToPHWO4I4Transposed()
129 if (out.size() != GetElementsSizeForPHWO4I4(shape)) { in ConvertToPHWO4I4()
132 out.size(), " != ", GetElementsSizeForPHWO4I4(shape))); in ConvertToPHWO4I4()
164 std::vector<float> transposed(GetElementsSizeForPHWO4I4(tensor.shape)); in ConvertToPHWO4I4()
Dconvert.h67 uint32_t GetElementsSizeForPHWO4I4(const OHWI& shape);
82 uint32_t GetElementsSizeForPHWO4I4(const IHWO& shape);