Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dconvert.h82 uint32_t GetElementsSizeForPHWO4I4(const IHWO& shape);
85 absl::Status ConvertToPHWO4I4(absl::Span<const float> in, const IHWO& shape,
90 const Tensor<IHWO, DataType::FLOAT32>& tensor);
Dshape.h56 IHWO = 8, enumerator
203 using IHWO = StrongShape<Layout::IHWO>; variable
362 TFLITE_GPU_LAYOUT_TRAITS(IHWO, Axis::INPUT_CHANNELS, Axis::HEIGHT, Axis::WIDTH,
609 case Layout::IHWO:
610 return f.template operator()<Layout::IHWO>();
Dshape.cc96 case Layout::IHWO: in ToString()
Dconvert.cc91 uint32_t GetElementsSizeForPHWO4I4(const IHWO& shape) { in GetElementsSizeForPHWO4I4()
122 absl::Status ConvertToPHWO4I4(absl::Span<const float> in, const IHWO& shape, in ConvertToPHWO4I4()
163 const Tensor<IHWO, DataType::FLOAT32>& tensor) { in ConvertToPHWO4I4() argument