Lines Matching refs:inputShape
36 bool compute(const InputType* inputData, const Shape& inputShape, OutputType* outputData) { in compute() argument
37 const int numElements = getNumberOfElements(inputShape); in compute()
38 const int32_t zeroPoint = inputShape.offset; in compute()
39 const float scale = inputShape.scale; in compute()
48 bool computePerChannel(const int8_t* inputData, const Shape& inputShape, OutputType* outputData) { in computePerChannel() argument
53 std::get<Operand::SymmPerChannelQuantParams>(inputShape.extraParams).channelDim; in computePerChannel()
55 for (int i = getNumberOfDimensions(inputShape) - 1; i > channelDim; --i) { in computePerChannel()
56 stride *= getSizeOfDimension(inputShape, i); in computePerChannel()
59 const int numElements = getNumberOfElements(inputShape); in computePerChannel()
60 const int32_t zeroPoint = inputShape.offset; in computePerChannel()
67 const int scaleIndex = (i / stride) % getSizeOfDimension(inputShape, channelDim); in computePerChannel()
68 const float scale = std::get<Operand::SymmPerChannelQuantParams>(inputShape.extraParams) in computePerChannel()
121 const Shape& inputShape = context->getInputShape(kInputTensor); in execute() local
125 return compute(inputBuffer, inputShape, in execute()
128 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
133 return compute(inputBuffer, inputShape, in execute()
136 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
141 return compute(inputBuffer, inputShape, in execute()
144 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
149 return computePerChannel(inputBuffer, inputShape, in execute()
152 return computePerChannel(inputBuffer, inputShape, in execute()