Searched refs:extOutputShape (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Reshape.cpp | 119 const auto extOutputShape = in padGeneric() local
122 const int outputBatch = extOutputShape.Dims(0); in padGeneric()
123 const int outputHeight = extOutputShape.Dims(1); in padGeneric()
124 const int outputWidth = extOutputShape.Dims(2); in padGeneric()
125 const int outputDepth = extOutputShape.Dims(3); in padGeneric()
138 outputData + tflite::Offset(extOutputShape, outB, 0, 0, 0), padValue, in padGeneric()
144 outputData + tflite::Offset(extOutputShape, outB, outH, 0, 0), padValue, in padGeneric()
150 outputData + tflite::Offset(extOutputShape, outB, outH, outW, 0), in padGeneric()
155 outputData + tflite::Offset(extOutputShape, outB, outH, outW, leftDPadding); in padGeneric()
163 outputData + tflite::Offset(extOutputShape, outB, outH, outW, in padGeneric()
[all …]
|