Lines Matching refs:inputShape
101 uint32_t numBatches = getSizeOfDimension(inputShape, 0); \
102 uint32_t inputHeight = getSizeOfDimension(inputShape, 1); \
103 uint32_t inputWidth = getSizeOfDimension(inputShape, 2); \
104 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); \
115 bool transposeConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in transposeConvNhwc() argument
171 bool transposeConvNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in transposeConvNhwc() argument
192 int32_t inputOffset = -inputShape.offset; in transposeConvNhwc()
199 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in transposeConvNhwc()
266 bool transposeConvNhwc(const _Float16* inputData, const Shape& inputShape, in transposeConvNhwc() argument
272 std::vector<float> inputData_float32(getNumberOfElements(inputShape)); in transposeConvNhwc()
281 transposeConvNhwc(inputData_float32.data(), inputShape, filterData_float32.data(), filterShape, in transposeConvNhwc()
290 bool transposeConv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in transposeConv() argument
296 NN_RET_CHECK(input.initialize(inputData, inputShape)); in transposeConv()
305 bool transposeConvQuant8PerChannelNhwc(const uint8_t* inputData, const Shape& inputShape, in transposeConvQuant8PerChannelNhwc() argument
327 int32_t inputOffset = -inputShape.offset; in transposeConvQuant8PerChannelNhwc()
337 biasChannelShape.scale = filterScales[i] * inputShape.scale; in transposeConvQuant8PerChannelNhwc()
340 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in transposeConvQuant8PerChannelNhwc()
407 bool transposeConvQuant8PerChannel(const uint8_t* inputData, const Shape& inputShape, in transposeConvQuant8PerChannel() argument
414 NN_RET_CHECK(input.initialize(inputData, inputShape)); in transposeConvQuant8PerChannel()