Lines Matching refs:Shape
32 bool floorFloat16(const _Float16* inputData, _Float16* outputData, const Shape& shape) { in floorFloat16()
43 bool floorFloat32(const float* inputData, float* outputData, const Shape& shape) { in floorFloat32()
51 bool meanFloat16(_Float16* inputData, const Shape& inputShape, const int32_t* axis, in meanFloat16()
52 const Shape& axisShape, bool keepDims, _Float16* outputData, in meanFloat16()
53 const Shape& outputShape) { in meanFloat16()
66 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape, in meanGeneric()
67 bool keepDims, T* outputData, const Shape& outputShape) { in meanGeneric()
95 template bool meanGeneric<float, float>(float* inputData, const Shape& inputShape,
96 const int32_t* axis, const Shape& axisShape, bool keepDims,
97 float* outputData, const Shape& outputShape);
98 template bool meanGeneric<uint8_t, int32_t>(uint8_t* inputData, const Shape& inputShape,
99 const int32_t* axis, const Shape& axisShape,
101 const Shape& outputShape);