Lines Matching refs:outputShape

35               const Shape& outputShape) {  in copyData()  argument
44 T* outputData, const Shape& outputShape) { in depthToSpaceGeneric() argument
47 outputData, convertShapeToDims(outputShape)); in depthToSpaceGeneric()
52 const Shape& outputShape);
55 const Shape& outputShape);
58 const Shape& outputShape);
61 const Shape& outputShape);
65 T* outputData, const Shape& outputShape) { in spaceToDepthGeneric() argument
68 outputData, convertShapeToDims(outputShape)); in spaceToDepthGeneric()
73 const Shape& outputShape);
76 const Shape& outputShape);
79 const Shape& outputShape);
82 const Shape& outputShape);
86 T* outputData, const Shape& outputShape) { in padGeneric() argument
114 tflite::RuntimeShape::ExtendedShape(4, convertShapeToTflshape(outputShape)); in padGeneric()
186 const Shape& outputShape);
189 const Shape& outputShape);
192 const Shape& outputShape);
195 const Shape& outputShape);
199 T* outputData, const Shape& outputShape) { in batchToSpaceGeneric() argument
206 convertShapeToDims(outputShape)); in batchToSpaceGeneric()
211 const Shape& outputShape);
214 const Shape& outputShape);
217 const Shape& outputShape);
220 const Shape& outputShape);
225 const Shape& outputShape) { in spaceToBatchGeneric() argument
230 {.output_offset = outputShape.offset}, convertShapeToTflshape(inputShape), inputData, in spaceToBatchGeneric()
232 convertShapeToTflshape(outputShape), outputData); in spaceToBatchGeneric()
238 const Shape& outputShape);
242 const Shape& outputShape);
246 const Shape& outputShape);
250 const Shape& outputShape);