• Home
  • Raw
  • Download

Lines Matching refs:axis

53                                          int32_t axis, float* outputData,  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()
83 T beta, int32_t axis, T* outputData, const Shape& outputShape);
87 float bias, float alpha, float beta, int32_t axis, float* outputData, in localResponseNorm() argument
90 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in localResponseNorm()
91 radius = std::min(radius, static_cast<int32_t>(inputShape.dimensions[axis])); in localResponseNorm()
93 if (axis == ndim - 1) { in localResponseNorm()
102 return localResponseNormFloat32Impl(inputData, inputShape, radius, bias, alpha, beta, axis, in localResponseNorm()
109 _Float16 bias, _Float16 alpha, _Float16 beta, int32_t axis, in localResponseNorm() argument
116 localResponseNorm<float>(inputDataFloat32.data(), inputShape, radius, bias, alpha, beta, axis, in localResponseNorm()
125 int32_t axis = context->getNumInputs() == kNumInputs in executeTyped() local
128 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in executeTyped()
132 context->getInputValue<T>(kAlphaScalar), context->getInputValue<T>(kBetaScalar), axis, in executeTyped()
186 int32_t axis = context->getNumInputs() == kNumInputs in prepare() local
190 NN_RET_CHECK_GE(axis, -numDimensions); in prepare()
191 NN_RET_CHECK_LT(axis, numDimensions); in prepare()