Lines Matching refs:Shape
33 bool meanFloat16(_Float16* inputData, const Shape& inputShape, const int32_t* axis, in meanFloat16()
34 const Shape& axisShape, bool keepDims, _Float16* outputData, in meanFloat16()
35 const Shape& outputShape) { in meanFloat16()
48 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape, in meanGeneric()
49 bool keepDims, T* outputData, const Shape& outputShape) { in meanGeneric()
77 template bool meanGeneric<float, float>(float* inputData, const Shape& inputShape,
78 const int32_t* axis, const Shape& axisShape, bool keepDims,
79 float* outputData, const Shape& outputShape);
80 template bool meanGeneric<uint8_t, int32_t>(uint8_t* inputData, const Shape& inputShape,
81 const int32_t* axis, const Shape& axisShape,
83 const Shape& outputShape);
84 template bool meanGeneric<int8_t, int32_t>(int8_t* inputData, const Shape& inputShape,
85 const int32_t* axis, const Shape& axisShape,
87 const Shape& outputShape);