Lines Matching refs:axis

84                               float bias, float alpha, float beta, int32_t axis,
87                               float bias, float alpha, float beta, int32_t axis, float* outputData,
113 bool meanFloat16(_Float16* inputData, const Shape& inputShape, const int32_t* axis,
117 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape,
125 bool argMinMaxGeneric(const uint8_t* inputData, const Shape& inputShape, int32_t axis,
128 bool splitFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis,
132 bool splitFloat32(const float* inputData, const Shape& inputShape, const int32_t axis,
136 bool splitInt32(const int32_t* inputData, const Shape& inputShape, const int32_t axis,
140 bool splitQuant8(const uint8_t* inputData, const Shape& inputShape, const int32_t axis,
144 bool splitQuant8Signed(const int8_t* inputData, const Shape& inputShape, const int32_t axis,
180                            int32_t axis, uint8_t* outputData, const Shape& outputShape);