Searched refs:nchw (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/tools/test_generator/tests/P_naming/ |
| D | conv_float.mod.py | 41 nchw = DataLayoutConverter("NCHW", name="nchw_layout").Identify([i1, f1, output], [layout]) variable
54 nchw, defaultName="nhwc"
|
| /frameworks/ml/nn/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
54 nchw, defaultName="nhwc"
|
| /frameworks/ml/nn/tools/test_generator/ |
| D | README.md | 203 …nchw], [default, relaxed, quant8]]`. This will result in 6 examples: `[default, default], [default…
208 # Add two groups of variations [default, nchw] and [default, relaxed, quant8]
209 example.AddVariations(nchw).AddVariations(relaxed, quant8)
215 example.AddVariations(nchw, defaultName="nhwc").AddVariations(relaxed, quant8)
221 # Add two groups of variations [nchw] and [default, relaxed, quant8]
222 example.AddVariations(nchw, includeDefault=False).AddVariations(relaxed, quant8)
225 The example above will result in 3 examples: `[nchw, default], [nchw, relaxed], [nchw, quant8]`.
265 example.AddVariations(DataLayoutConverter("nchw").Identify(op_list))
266 example.AddVariations(("nchw", op_list))
|
| /frameworks/ml/nn/common/include/ |
| D | CpuOperationUtils.h | 117 inline bool convertNchwToNhwc(const T* nchw, const Shape& nchwShape, std::vector<T>* nhwc, in convertNchwToNhwc() argument
131 *to++ = nchw[fromIndex]; in convertNchwToNhwc()
139 inline bool convertNhwcToNchw(const std::vector<T>& nhwc, const Shape& nhwcShape, T* nchw) { in convertNhwcToNchw() argument
149 *nchw++ = from[fromIndex]; in convertNhwcToNchw()
|