48 void CalculateActivationRangeImpl(int32_t activation, const Shape& outputShape, int32_t qmin,  in CalculateActivationRangeImpl()  argument
50     const auto scale = outputShape.scale;  in CalculateActivationRangeImpl()
51     const auto zero_point = outputShape.offset;  in CalculateActivationRangeImpl()
257                                       const Shape& biasShape, const Shape& outputShape,  in GetQuantizedConvolutionMultipler()  argument
267     *multiplier = input_product_scale / outputShape.scale;  in GetQuantizedConvolutionMultipler()
271 void CalculateActivationRangeUint8(int32_t activation, const Shape& outputShape, int32_t* act_min,  in CalculateActivationRangeUint8()  argument
276     CalculateActivationRangeImpl(activation, outputShape, qmin, qmax, act_min, act_max);  in CalculateActivationRangeUint8()
279 void CalculateActivationRangeInt8(int32_t activation, const Shape& outputShape, int32_t* act_min,  in CalculateActivationRangeInt8()  argument
284     CalculateActivationRangeImpl(activation, outputShape, qmin, qmax, act_min, act_max);  in CalculateActivationRangeInt8()
460 bool embeddingLookupPrepare(const Shape& valueShape, const Shape& lookupShape, Shape* outputShape) {  in embeddingLookupPrepare()  argument
467     outputShape->type = valueShape.type;  in embeddingLookupPrepare()
468     outputShape->dimensions = {lookups, columns};  in embeddingLookupPrepare()
470         outputShape->dimensions.push_back(getSizeOfDimension(valueShape, i));  in embeddingLookupPrepare()
472     outputShape->offset = valueShape.offset;  in embeddingLookupPrepare()
473     outputShape->scale = valueShape.scale;  in embeddingLookupPrepare()
479                             const Shape& valueShape, Shape* outputShape, Shape* hitShape) {  in hashtableLookupPrepare()  argument
485     outputShape->type = valueShape.type;  in hashtableLookupPrepare()
486     outputShape->dimensions = {lookups};  in hashtableLookupPrepare()
488         outputShape->dimensions.push_back(getSizeOfDimension(valueShape, i));  in hashtableLookupPrepare()
490     outputShape->offset = valueShape.offset;  in hashtableLookupPrepare()
491     outputShape->scale = valueShape.scale;  in hashtableLookupPrepare()