Lines Matching refs:outputShape
50 void CalculateActivationRangeImpl(int32_t activation, const Shape& outputShape, int32_t qmin, in CalculateActivationRangeImpl() argument
52 const auto scale = outputShape.scale; in CalculateActivationRangeImpl()
53 const auto zero_point = outputShape.offset; in CalculateActivationRangeImpl()
259 const Shape& biasShape, const Shape& outputShape, in GetQuantizedConvolutionMultipler() argument
269 *multiplier = input_product_scale / outputShape.scale; in GetQuantizedConvolutionMultipler()
273 void CalculateActivationRangeUint8(int32_t activation, const Shape& outputShape, int32_t* act_min, in CalculateActivationRangeUint8() argument
278 CalculateActivationRangeImpl(activation, outputShape, qmin, qmax, act_min, act_max); in CalculateActivationRangeUint8()
281 void CalculateActivationRangeInt8(int32_t activation, const Shape& outputShape, int32_t* act_min, in CalculateActivationRangeInt8() argument
286 CalculateActivationRangeImpl(activation, outputShape, qmin, qmax, act_min, act_max); in CalculateActivationRangeInt8()
462 bool embeddingLookupPrepare(const Shape& valueShape, const Shape& lookupShape, Shape* outputShape) { in embeddingLookupPrepare() argument
469 outputShape->type = valueShape.type; in embeddingLookupPrepare()
470 outputShape->dimensions = {lookups, columns}; in embeddingLookupPrepare()
472 outputShape->dimensions.push_back(getSizeOfDimension(valueShape, i)); in embeddingLookupPrepare()
474 outputShape->offset = valueShape.offset; in embeddingLookupPrepare()
475 outputShape->scale = valueShape.scale; in embeddingLookupPrepare()
481 const Shape& valueShape, Shape* outputShape, Shape* hitShape) { in hashtableLookupPrepare() argument
487 outputShape->type = valueShape.type; in hashtableLookupPrepare()
488 outputShape->dimensions = {lookups}; in hashtableLookupPrepare()
490 outputShape->dimensions.push_back(getSizeOfDimension(valueShape, i)); in hashtableLookupPrepare()
492 outputShape->offset = valueShape.offset; in hashtableLookupPrepare()
493 outputShape->scale = valueShape.scale; in hashtableLookupPrepare()