Searched defs:axisShape (Results 1 – 2 of 2) sorted by relevance
52 const Shape& axisShape, bool keepDims, _Float16* outputData, in meanFloat16()66 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape, in meanGeneric()
696 const Shape& axisShape, in meanPrepare()