Searched defs:nhwc (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/packages/modules/NeuralNetworks/common/include/ |
| D | CpuOperationUtils.h | 118 inline bool convertNchwToNhwc(const T* nchw, const Shape& nchwShape, std::vector<T>* nhwc, in convertNchwToNhwc()
140 inline bool convertNhwcToNchw(const std::vector<T>& nhwc, const Shape& nhwcShape, T* nchw) { in convertNhwcToNchw()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cudnn/ |
| H A D | Descriptors.h | 52 static inline void fixSizeOneDimStride(int dim, const T *size, T *stride, bool nhwc) { in fixSizeOneDimStride()
167 void set(cudnnDataType_t dataType, int dim, int* size, int* stride, bool nhwc) { in set()
|
| H A D | Descriptors.cpp | 55 …:set(cudnnDataType_t datatype, IntArrayRef t_sizes, IntArrayRef t_strides, size_t pad, bool nhwc) { in set()
|
| /aosp_15_r20/external/armnn/samples/common/include/ArmnnUtils/ |
| H A D | ArmnnNetworkExecutor.hpp | 234 armnnUtils::DataLayoutIndexed nhwc(armnn::DataLayout::NHWC); in GetImageAspectRatio() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | xsmm_conv2d_test.cc | 51 LIBXSMM_INLINE void naive_copy_NCHW_to_NHWC(const float* nchw, Tensor& nhwc, in naive_copy_NCHW_to_NHWC()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nnapi.py | 20 def nhwc(t): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | arithmetic_optimizer_test.cc | 1463 Output nhwc = ops::Reshape( in TEST_F() local
|