Lines Matching refs:inputShape

34     uint32_t numBatches = getSizeOfDimension(inputShape, 0);    \
35 uint32_t inputHeight = getSizeOfDimension(inputShape, 1); \
36 uint32_t inputWidth = getSizeOfDimension(inputShape, 2); \
37 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); \
46 bool groupedConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData, in groupedConvFloat32() argument
104 bool groupedConvQuant8(const T* inputData, const Shape& inputShape, const T* filterData, in groupedConvQuant8() argument
113 int32_t inputOffset = -inputShape.offset; in groupedConvQuant8()
120 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in groupedConvQuant8()
181 template bool groupedConvQuant8<int8_t>(const int8_t* inputData, const Shape& inputShape,
190 template bool groupedConvQuant8<uint8_t>(const uint8_t* inputData, const Shape& inputShape,
200 bool groupedConvQuant8PerChannel(const T* inputData, const Shape& inputShape, in groupedConvQuant8PerChannel() argument
210 int32_t inputOffset = -inputShape.offset; in groupedConvQuant8PerChannel()
221 biasChannelShape.scale = filterScales[i] * inputShape.scale; in groupedConvQuant8PerChannel()
224 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in groupedConvQuant8PerChannel()
285 bool groupedConvFloat16(const _Float16* inputData, const Shape& inputShape, in groupedConvFloat16() argument
293 std::vector<float> inputData_float32(getNumberOfElements(inputShape)); in groupedConvFloat16()
302 groupedConvFloat32(inputData_float32.data(), inputShape, filterData_float32.data(), filterShape, in groupedConvFloat16()
312 const uint8_t* inputData, const Shape& inputShape, const int8_t* filterData,
319 const int8_t* inputData, const Shape& inputShape, const int8_t* filterData,