Lines Matching refs:Shape
56 bool fullyConnectedFloat32(const float* inputData, const Shape& inputShape, in fullyConnectedFloat32()
57 const float* weightsData, const Shape& weightsShape, in fullyConnectedFloat32()
58 const float* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat32()
59 float* outputData, const Shape& outputShape) { in fullyConnectedFloat32()
86 bool fullyConnectedFloat16(const _Float16* inputData, const Shape& inputShape, in fullyConnectedFloat16()
87 const _Float16* weightsData, const Shape& weightsShape, in fullyConnectedFloat16()
88 const _Float16* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat16()
89 _Float16* outputData, const Shape& outputShape) { in fullyConnectedFloat16()
107 bool fullyConnectedQuant8(const uint8_t* inputData, const Shape& inputShape, in fullyConnectedQuant8()
108 const uint8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8()
109 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8()
110 uint8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8()
148 bool fullyConnectedQuant8(const int8_t* inputData, const Shape& inputShape, in fullyConnectedQuant8()
149 const int8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8()
150 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8()
151 int8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8()
185 bool validateShapes(const Shape& input, const Shape& weights, const Shape& bias, in validateShapes()
186 Shape* output = nullptr) { in validateShapes()
284 Shape input = context->getInputShape(kInputTensor); in validate()
285 Shape weights = context->getInputShape(kWeightsTensor); in validate()
286 Shape bias = context->getInputShape(kBiasTensor); in validate()
296 Shape input = context->getInputShape(kInputTensor); in prepare()
297 Shape weights = context->getInputShape(kWeightsTensor); in prepare()
298 Shape bias = context->getInputShape(kBiasTensor); in prepare()
299 Shape output = context->getOutputShape(kOutputTensor); in prepare()