Lines Matching refs:outputData

32 bool copyData(const void* inputData, const Shape& inputShape, void* outputData,  in copyData()  argument
36 memcpy(outputData, inputData, count); in copyData()
42 T* outputData, const Shape& outputShape) { in depthToSpaceGeneric() argument
45 outputData, convertShapeToDims(outputShape)); in depthToSpaceGeneric()
49 int32_t blockSize, float* outputData,
52 int32_t blockSize, _Float16* outputData,
55 int32_t blockSize, uint8_t* outputData,
60 T* outputData, const Shape& outputShape) { in spaceToDepthGeneric() argument
63 outputData, convertShapeToDims(outputShape)); in spaceToDepthGeneric()
67 int32_t blockSize, float* outputData,
70 int32_t blockSize, _Float16* outputData,
73 int32_t blockSize, uint8_t* outputData,
78 T* outputData, const Shape& outputShape) { in padGeneric() argument
119 outputData, padValue, leftBPadding * outputHeight * outputWidth * outputDepth); in padGeneric()
124 outputData + tflite::Offset(extOutputShape, outB, 0, 0, 0), padValue, in padGeneric()
130 outputData + tflite::Offset(extOutputShape, outB, outH, 0, 0), padValue, in padGeneric()
136 outputData + tflite::Offset(extOutputShape, outB, outH, outW, 0), in padGeneric()
141 outputData + tflite::Offset(extOutputShape, outB, outH, outW, leftDPadding); in padGeneric()
149 outputData + tflite::Offset(extOutputShape, outB, outH, outW, in padGeneric()
156 outputData + tflite::Offset(extOutputShape, outB, outH, in padGeneric()
163 outputData + tflite::Offset(extOutputShape, outB, outputHeight - rightHPadding, in padGeneric()
170 outputData + tflite::Offset(extOutputShape, outputBatch - rightBPadding, 0, 0, 0), in padGeneric()
177 const int32_t* paddings, float padValue, float* outputData,
180 const int32_t* paddings, _Float16 padValue, _Float16* outputData,
183 const int32_t* paddings, uint8_t padValue, uint8_t* outputData,
188 T* outputData, const Shape& outputShape) { in batchToSpaceGeneric() argument
194 blockSizeDim, crops, cropsDim, outputData, in batchToSpaceGeneric()
199 const int32_t* blockSize, float* outputData,
202 const int32_t* blockSize, _Float16* outputData,
205 const int32_t* blockSize, uint8_t* outputData,
210 const int32_t* padding, const Shape& paddingShape, T* outputData, in spaceToBatchGeneric() argument
218 convertShapeToTflshape(outputShape), outputData); in spaceToBatchGeneric()
223 const Shape& paddingShape, float* outputData,
227 const Shape& paddingShape, _Float16* outputData,
231 const Shape& paddingShape, uint8_t* outputData,