Lines Matching refs:Shape
47 bool fullyConnectedFloat32(const float* inputData, const Shape& inputShape, in fullyConnectedFloat32()
48 const float* weightsData, const Shape& weightsShape, in fullyConnectedFloat32()
49 const float* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat32()
50 float* outputData, const Shape& outputShape) { in fullyConnectedFloat32()
77 bool fullyConnectedFloat16(const _Float16* inputData, const Shape& inputShape, in fullyConnectedFloat16()
78 const _Float16* weightsData, const Shape& weightsShape, in fullyConnectedFloat16()
79 const _Float16* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat16()
80 _Float16* outputData, const Shape& outputShape) { in fullyConnectedFloat16()
98 bool fullyConnectedQuant8(const uint8_t* inputData, const Shape& inputShape, in fullyConnectedQuant8()
99 const uint8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8()
100 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8()
101 uint8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8()
194 Shape input = context->getInputShape(kInputTensor); in prepare()
195 Shape weights = context->getInputShape(kWeightsTensor); in prepare()
196 Shape bias = context->getInputShape(kBiasTensor); in prepare()
220 Shape output = context->getOutputShape(kOutputTensor); in prepare()