Searched refs:IHWO (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | convert.h | 82 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);
|
| D | shape.h | 56 IHWO = 8, enumerator
204 using IHWO = StrongShape<Layout::IHWO>; variable
364 TFLITE_GPU_LAYOUT_TRAITS(IHWO, Axis::INPUT_CHANNELS, Axis::HEIGHT, Axis::WIDTH,
613 case Layout::IHWO:
614 return f.template operator()<Layout::IHWO>();
|
| D | shape.cc | 96 case Layout::IHWO: in ToString()
|
| D | convert.cc | 91 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
|