Searched refs:outShape (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 499 Shape outShape = from.shape(); in convertFromNhwc() local
501 outShape.dimensions = {fromDim[0], fromDim[3], fromDim[1], fromDim[2]}; in convertFromNhwc()
503 if (!setInfoAndAllocateIfNeeded(&to, outShape, result)) { in convertFromNhwc()
523 Shape outShape = from.shape(); in convertFromNhwc() local
526 if (!setInfoAndAllocateIfNeeded(&to, outShape, result)) { in convertFromNhwc()
727 Shape outShape = output.shape(); in executeOperation() local
729 if (!floorPrepare(input.shape(), &outShape) || in executeOperation()
730 !setInfoAndAllocateIfNeeded(&output, outShape, &result)) { in executeOperation()
735 reinterpret_cast<float*>(output.buffer), outShape); in executeOperation()
738 reinterpret_cast<_Float16*>(output.buffer), outShape); in executeOperation()
[all …]
|