Lines Matching refs:outputData
34 bool copyData(const void* inputData, const Shape& inputShape, void* outputData, in copyData() argument
38 memcpy(outputData, inputData, count); in copyData()
44 T* outputData, const Shape& outputShape) { in depthToSpaceGeneric() argument
47 outputData, convertShapeToDims(outputShape)); in depthToSpaceGeneric()
51 int32_t blockSize, float* outputData,
54 int32_t blockSize, _Float16* outputData,
57 int32_t blockSize, uint8_t* outputData,
60 int32_t blockSize, int8_t* outputData,
65 T* outputData, const Shape& outputShape) { in spaceToDepthGeneric() argument
68 outputData, convertShapeToDims(outputShape)); in spaceToDepthGeneric()
72 int32_t blockSize, float* outputData,
75 int32_t blockSize, _Float16* outputData,
78 int32_t blockSize, uint8_t* outputData,
81 int32_t blockSize, int8_t* outputData,
86 T* outputData, const Shape& outputShape) { in padGeneric() argument
127 outputData, padValue, leftBPadding * outputHeight * outputWidth * outputDepth); 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()
164 outputData + tflite::Offset(extOutputShape, outB, outH, in padGeneric()
171 outputData + tflite::Offset(extOutputShape, outB, outputHeight - rightHPadding, in padGeneric()
178 outputData + tflite::Offset(extOutputShape, outputBatch - rightBPadding, 0, 0, 0), in padGeneric()
185 const int32_t* paddings, float padValue, float* outputData,
188 const int32_t* paddings, _Float16 padValue, _Float16* outputData,
191 const int32_t* paddings, uint8_t padValue, uint8_t* outputData,
194 const int32_t* paddings, int8_t padValue, int8_t* outputData,
199 T* outputData, const Shape& outputShape) { in batchToSpaceGeneric() argument
205 blockSizeDim, crops, cropsDim, outputData, in batchToSpaceGeneric()
210 const int32_t* blockSize, float* outputData,
213 const int32_t* blockSize, _Float16* outputData,
216 const int32_t* blockSize, uint8_t* outputData,
219 const int32_t* blockSize, int8_t* outputData,
224 const int32_t* padding, const Shape& paddingShape, T* outputData, in spaceToBatchGeneric() argument
232 convertShapeToTflshape(outputShape), outputData); in spaceToBatchGeneric()
237 const Shape& paddingShape, float* outputData,
241 const Shape& paddingShape, _Float16* outputData,
245 const Shape& paddingShape, uint8_t* outputData,
249 const Shape& paddingShape, int8_t* outputData,