Searched defs:nhwc (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/aten/src/ATen/cudnn/ |
| 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()
|
| D | Descriptors.cpp | 55 …:set(cudnnDataType_t datatype, IntArrayRef t_sizes, IntArrayRef t_strides, size_t pad, bool nhwc) { in set()
|
| /external/armnn/samples/common/include/ArmnnUtils/ |
| D | ArmnnNetworkExecutor.hpp | 234 armnnUtils::DataLayoutIndexed nhwc(armnn::DataLayout::NHWC); in GetImageAspectRatio() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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()
|
| /external/pytorch/test/ |
| D | test_nnapi.py | 20 def nhwc(t): function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 1463 Output nhwc = ops::Reshape( in TEST_F() local
|