Searched refs:outShape (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 569 Shape outShape = from.shape(); in convertFromNhwc() local
571 outShape.dimensions = {fromDim[0], fromDim[3], fromDim[1], fromDim[2]}; in convertFromNhwc()
573 if (!setInfoAndAllocateIfNeeded(&to, outShape, result)) { in convertFromNhwc()
597 Shape outShape = from.shape(); in convertFromNhwc() local
600 if (!setInfoAndAllocateIfNeeded(&to, outShape, result)) { in convertFromNhwc()
858 Shape outShape = output.shape(); in executeOperation() local
862 getNumberOfElements(targetShape.shape()), &outShape) && in executeOperation()
863 setInfoAndAllocateIfNeeded(&output, outShape, &result) && in executeOperation()
864 copyData(input.buffer, input.shape(), output.buffer, outShape); in executeOperation()
876 Shape outShape = output.shape(); in executeOperation() local
[all …]
|