Lines Matching refs:inputShape
47 bool fullyConnectedFloat32(const float* inputData, const Shape& inputShape, in fullyConnectedFloat32() argument
58 uint32_t input_n_elements = getNumberOfElements(inputShape); in fullyConnectedFloat32()
61 tflite::reference_ops::FullyConnected(inputData, convertShapeToDims(inputShape), in fullyConnectedFloat32()
68 tflite::optimized_ops::FullyConnected(inputData, convertShapeToDims(inputShape), in fullyConnectedFloat32()
77 bool fullyConnectedFloat16(const _Float16* inputData, const Shape& inputShape, in fullyConnectedFloat16() argument
82 std::vector<float> inputDataFloat32(getNumberOfElements(inputShape)); in fullyConnectedFloat16()
90 fullyConnectedFloat32(inputDataFloat32.data(), inputShape, weightsDataFloat32.data(), in fullyConnectedFloat16()
98 bool fullyConnectedQuant8(const uint8_t* inputData, const Shape& inputShape, in fullyConnectedQuant8() argument
103 int32_t inputOffset = -inputShape.offset; in fullyConnectedQuant8()
113 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, weightsShape, biasShape, outputShape, in fullyConnectedQuant8()
129 tflite::optimized_ops::FullyConnected(inputData, convertShapeToDims(inputShape), inputOffset, in fullyConnectedQuant8()