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 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>();
|
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
|