Searched refs:inShape (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Elementwise.cpp | 124 const Shape inShape = context->getInputShape(kInputTensor); in executeRsqrt() local
127 makeQuantized(frsqrt, inShape.scale, static_cast<uint8_t>(inShape.offset), in executeRsqrt()
134 const Shape inShape = context->getInputShape(kInputTensor); in executeRsqrt() local
137 makeQuantized(frsqrt, inShape.scale, static_cast<int8_t>(inShape.offset), in executeRsqrt()
|
| /packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 444 Shape inShape = from.shape(); in convertToNhwc() local
446 inShape.dimensions = {fromDim[0], fromDim[2], fromDim[3], fromDim[1]}; in convertToNhwc()
449 if (!setInfoAndAllocateIfNeeded(&to, inShape, &result)) { in convertToNhwc()
|