Lines Matching refs:Shape
34 bool copyData(const void* inputData, const Shape& inputShape, void* outputData, in copyData()
35 const Shape& outputShape) { in copyData()
43 bool depthToSpaceGeneric(const T* inputData, const Shape& inputShape, int32_t blockSize, in depthToSpaceGeneric()
44 T* outputData, const Shape& outputShape) { in depthToSpaceGeneric()
50 template bool depthToSpaceGeneric<float>(const float* inputData, const Shape& inputShape,
52 const Shape& outputShape);
53 template bool depthToSpaceGeneric<_Float16>(const _Float16* inputData, const Shape& inputShape,
55 const Shape& outputShape);
56 template bool depthToSpaceGeneric<uint8_t>(const uint8_t* inputData, const Shape& inputShape,
58 const Shape& outputShape);
59 template bool depthToSpaceGeneric<int8_t>(const int8_t* inputData, const Shape& inputShape,
61 const Shape& outputShape);
64 bool spaceToDepthGeneric(const T* inputData, const Shape& inputShape, int32_t blockSize, in spaceToDepthGeneric()
65 T* outputData, const Shape& outputShape) { in spaceToDepthGeneric()
71 template bool spaceToDepthGeneric<float>(const float* inputData, const Shape& inputShape,
73 const Shape& outputShape);
74 template bool spaceToDepthGeneric<_Float16>(const _Float16* inputData, const Shape& inputShape,
76 const Shape& outputShape);
77 template bool spaceToDepthGeneric<uint8_t>(const uint8_t* inputData, const Shape& inputShape,
79 const Shape& outputShape);
80 template bool spaceToDepthGeneric<int8_t>(const int8_t* inputData, const Shape& inputShape,
82 const Shape& outputShape);
85 bool padGeneric(const T* inputData, const Shape& inputShape, const int32_t* paddings, T padValue, in padGeneric()
86 T* outputData, const Shape& outputShape) { in padGeneric()
184 template bool padGeneric<float>(const float* inputData, const Shape& inputShape,
186 const Shape& outputShape);
187 template bool padGeneric<_Float16>(const _Float16* inputData, const Shape& inputShape,
189 const Shape& outputShape);
190 template bool padGeneric<uint8_t>(const uint8_t* inputData, const Shape& inputShape,
192 const Shape& outputShape);
193 template bool padGeneric<int8_t>(const int8_t* inputData, const Shape& inputShape,
195 const Shape& outputShape);
198 bool batchToSpaceGeneric(const T* inputData, const Shape& inputShape, const int32_t* blockSize, in batchToSpaceGeneric()
199 T* outputData, const Shape& outputShape) { in batchToSpaceGeneric()
209 template bool batchToSpaceGeneric<float>(const float* inputData, const Shape& inputShape,
211 const Shape& outputShape);
212 template bool batchToSpaceGeneric<_Float16>(const _Float16* inputData, const Shape& inputShape,
214 const Shape& outputShape);
215 template bool batchToSpaceGeneric<uint8_t>(const uint8_t* inputData, const Shape& inputShape,
217 const Shape& outputShape);
218 template bool batchToSpaceGeneric<int8_t>(const int8_t* inputData, const Shape& inputShape,
220 const Shape& outputShape);
223 bool spaceToBatchGeneric(const T* inputData, const Shape& inputShape, const int32_t* blockSize, in spaceToBatchGeneric()
224 const int32_t* padding, const Shape& paddingShape, T* outputData, in spaceToBatchGeneric()
225 const Shape& outputShape) { in spaceToBatchGeneric()
235 template bool spaceToBatchGeneric<float>(const float* inputData, const Shape& inputShape,
237 const Shape& paddingShape, float* outputData,
238 const Shape& outputShape);
239 template bool spaceToBatchGeneric<_Float16>(const _Float16* inputData, const Shape& inputShape,
241 const Shape& paddingShape, _Float16* outputData,
242 const Shape& outputShape);
243 template bool spaceToBatchGeneric<uint8_t>(const uint8_t* inputData, const Shape& inputShape,
245 const Shape& paddingShape, uint8_t* outputData,
246 const Shape& outputShape);
247 template bool spaceToBatchGeneric<int8_t>(const int8_t* inputData, const Shape& inputShape,
249 const Shape& paddingShape, int8_t* outputData,
250 const Shape& outputShape);