Lines Matching refs:inputShape
51 inline bool localResponseNormFloat32Impl(const float* inputData, const Shape& inputShape, in localResponseNormFloat32Impl() argument
56 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in localResponseNormFloat32Impl()
57 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in localResponseNormFloat32Impl()
59 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in localResponseNormFloat32Impl()
82 bool localResponseNorm(const T* inputData, const Shape& inputShape, int32_t radius, T bias, T alpha,
86 bool localResponseNorm<float>(const float* inputData, const Shape& inputShape, int32_t radius,
89 int32_t ndim = getNumberOfDimensions(inputShape);
90 NN_CHECK(handleNegativeAxis(inputShape, &axis));
91 radius = std::min(radius, static_cast<int32_t>(inputShape.dimensions[axis]));
98 param, convertShapeToTflshape(inputShape), inputData,
102 return localResponseNormFloat32Impl(inputData, inputShape, radius, bias, alpha, beta, axis,
108 bool localResponseNorm<_Float16>(const _Float16* inputData, const Shape& inputShape, int32_t radius,
112 std::vector<float> inputDataFloat32(getNumberOfElements(inputShape));
116 localResponseNorm<float>(inputDataFloat32.data(), inputShape, radius, bias, alpha, beta, axis,