Lines Matching refs:inputShape
29 inline bool localResponseNormFloat32Impl(const float* inputData, const Shape& inputShape, in localResponseNormFloat32Impl() argument
34 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in localResponseNormFloat32Impl()
35 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in localResponseNormFloat32Impl()
37 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in localResponseNormFloat32Impl()
59 bool localResponseNormFloat16(const _Float16* inputData, const Shape& inputShape, int32_t radius, in localResponseNormFloat16() argument
63 std::vector<float> inputDataFloat32(getNumberOfElements(inputShape)); in localResponseNormFloat16()
67 localResponseNormFloat32(inputDataFloat32.data(), inputShape, radius, bias, alpha, beta, axis, in localResponseNormFloat16()
74 bool localResponseNormFloat32(const float* inputData, const Shape& inputShape, int32_t radius, in localResponseNormFloat32() argument
77 int32_t ndim = getNumberOfDimensions(inputShape); in localResponseNormFloat32()
78 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in localResponseNormFloat32()
85 param, convertShapeToTflshape(inputShape), inputData, in localResponseNormFloat32()
89 return localResponseNormFloat32Impl(inputData, inputShape, radius, bias, alpha, beta, axis, in localResponseNormFloat32()