Searched defs:nchw (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/tools/test_generator/tests/P_naming/ |
| D | conv_float.mod.py | 41 nchw = DataLayoutConverter("NCHW", name="nchw_layout").Identify([i1, f1, output], [layout]) variable
|
| /packages/modules/NeuralNetworks/tools/test_generator/tests/P_vts_naming/ |
| D | conv_float.mod.py | 41 nchw = DataLayoutConverter("NCHW", name="nchw_layout").Identify([i1, f1, output], [layout]) variable
|
| /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()
|