Searched refs:extOutputShape (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | Reshape.cpp | 113 const auto extOutputShape = in padGeneric() local
116 const int outputBatch = extOutputShape.Dims(0); in padGeneric()
117 const int outputHeight = extOutputShape.Dims(1); in padGeneric()
118 const int outputWidth = extOutputShape.Dims(2); in padGeneric()
119 const int outputDepth = extOutputShape.Dims(3); in padGeneric()
132 outputData + tflite::Offset(extOutputShape, outB, 0, 0, 0), padValue, in padGeneric()
138 outputData + tflite::Offset(extOutputShape, outB, outH, 0, 0), padValue, in padGeneric()
144 outputData + tflite::Offset(extOutputShape, outB, outH, outW, 0), in padGeneric()
149 outputData + tflite::Offset(extOutputShape, outB, outH, outW, leftDPadding); in padGeneric()
157 outputData + tflite::Offset(extOutputShape, outB, outH, outW, in padGeneric()
[all …]
|