Lines Matching refs:inputShape
96 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
98 int32_t input_width = getSizeOfDimension(inputShape, useNchw ? 3 : 2); in initialize()
99 int32_t input_height = getSizeOfDimension(inputShape, useNchw ? 2 : 1); in initialize()
123 uint32_t height = getSizeOfDimension(inputShape, 1); \
124 uint32_t width = getSizeOfDimension(inputShape, 2); \
133 bool depthwiseConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in depthwiseConvNhwc() argument
159 tflite::reference_ops::DepthwiseConv(params, convertShapeToTflshape(inputShape), inputData, in depthwiseConvNhwc()
167 bool depthwiseConvNhwc(const _Float16* inputData, const Shape& inputShape, in depthwiseConvNhwc() argument
175 std::vector<float> inputDataFloat32(getNumberOfElements(inputShape)); in depthwiseConvNhwc()
183 depthwiseConvNhwc(inputDataFloat32.data(), inputShape, filterDataFloat32.data(), filterShape, in depthwiseConvNhwc()
193 bool depthwiseConvNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in depthwiseConvNhwc() argument
210 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in depthwiseConvNhwc()
226 .input_offset = -inputShape.offset, in depthwiseConvNhwc()
235 tflite::reference_ops::DepthwiseConv(params, convertShapeToTflshape(inputShape), inputData, in depthwiseConvNhwc()
244 bool depthwiseConvNhwc(const int8_t* inputData, Shape inputShape, const int8_t* filterData, in depthwiseConvNhwc() argument
253 std::vector<uint8_t> unsignedInput(getNumberOfElements(inputShape)); in depthwiseConvNhwc()
255 inputShape.offset += 128; in depthwiseConvNhwc()
264 NN_RET_CHECK(depthwiseConvNhwc(unsignedInput.data(), inputShape, unsignedFilter.data(), in depthwiseConvNhwc()
277 const T* inputData, const Shape& inputShape, const int8_t* filterData, in depthwiseConvQuant8PerChannelNhwc() argument
289 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in depthwiseConvQuant8PerChannelNhwc()
290 uint32_t inputHeight = getSizeOfDimension(inputShape, 1); in depthwiseConvQuant8PerChannelNhwc()
291 uint32_t inputWidth = getSizeOfDimension(inputShape, 2); in depthwiseConvQuant8PerChannelNhwc()
292 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); in depthwiseConvQuant8PerChannelNhwc()
300 int32_t inputOffset = -inputShape.offset; in depthwiseConvQuant8PerChannelNhwc()
311 biasChannelShape.scale = filterScales[i] * inputShape.scale; in depthwiseConvQuant8PerChannelNhwc()
313 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in depthwiseConvQuant8PerChannelNhwc()
373 bool depthwiseConv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in depthwiseConv() argument
382 NN_RET_CHECK(input.initialize(inputData, inputShape)); in depthwiseConv()
394 bool depthwiseConvQuant8PerChannel(const T* inputData, const Shape& inputShape, in depthwiseConvQuant8PerChannel() argument
405 NN_RET_CHECK(input.initialize(inputData, inputShape)); in depthwiseConvQuant8PerChannel()