Home
last modified time | relevance | path

Searched refs:QuantizationType (Results 1 – 14 of 14) sorted by relevance

/external/libtextclassifier/native/lang_id/common/
Dembedding-network-params.h28 enum class QuantizationType { enum
47 QuantizationType ParseQuantizationType(const std::string &s);
87 QuantizationType quant_type = QuantizationType::NONE;
138 matrix.quant_type = QuantizationType::NONE; in GetHiddenLayerBias()
174 matrix.quant_type = QuantizationType::NONE; in GetSoftmaxBias()
212 virtual QuantizationType embeddings_quant_type(int i) const { in embeddings_quant_type()
213 return QuantizationType::NONE; in embeddings_quant_type()
233 virtual QuantizationType hidden_weights_quant_type(int i) const { in hidden_weights_quant_type()
234 return QuantizationType::NONE; in hidden_weights_quant_type()
267 virtual QuantizationType softmax_weights_quant_type(int i) const { in softmax_weights_quant_type()
[all …]
Dembedding-network-params.cc25 QuantizationType ParseQuantizationType(const std::string &s) { in ParseQuantizationType()
27 return QuantizationType::NONE; in ParseQuantizationType()
30 return QuantizationType::UINT8; in ParseQuantizationType()
33 return QuantizationType::UINT4; in ParseQuantizationType()
36 return QuantizationType::FLOAT16; in ParseQuantizationType()
43 return QuantizationType::NONE; in ParseQuantizationType()
Dembedding-network.cc27 SAFTM_CHECK_EQ(static_cast<int>(QuantizationType::NONE), in CheckNoQuantization()
34 QuantizationType quant_type = matrix.quant_type; in GetMatrixRowSizeInBytes()
36 case QuantizationType::NONE: in GetMatrixRowSizeInBytes()
38 case QuantizationType::UINT8: in GetMatrixRowSizeInBytes()
40 case QuantizationType::UINT4: in GetMatrixRowSizeInBytes()
43 case QuantizationType::FLOAT16: in GetMatrixRowSizeInBytes()
93 case QuantizationType::NONE: { in SparseReluProductPlusBias()
121 case QuantizationType::FLOAT16: { in SparseReluProductPlusBias()
198 case QuantizationType::NONE: { in ConcatEmbeddings()
206 case QuantizationType::UINT8: { in ConcatEmbeddings()
[all …]
/external/libtextclassifier/native/lang_id/common/flatbuffers/
Dembedding-network-params-from-flatbuffer.cc86 case QuantizationType::NONE: in VerifyMatrix()
89 case QuantizationType::UINT8: { in VerifyMatrix()
96 case QuantizationType::UINT4: { in VerifyMatrix()
107 case QuantizationType::FLOAT16: { in VerifyMatrix()
402 QuantizationType EmbeddingNetworkParamsFromFlatbuffer::SafeGetQuantizationType( in SafeGetQuantizationType()
405 return QuantizationType::NONE; in SafeGetQuantizationType()
407 saft_fbs::QuantizationType quantization_type = matrix->quantization_type(); in SafeGetQuantizationType()
413 return QuantizationType::NONE; in SafeGetQuantizationType()
415 return QuantizationType::UINT8; in SafeGetQuantizationType()
417 return QuantizationType::UINT4; in SafeGetQuantizationType()
[all …]
Dembedding-network-params-from-flatbuffer.h77 QuantizationType embeddings_quant_type(int i) const override { in embeddings_quant_type()
102 QuantizationType hidden_weights_quant_type(int i) const override { in hidden_weights_quant_type()
141 QuantizationType softmax_weights_quant_type(int i) const override { in softmax_weights_quant_type()
259 QuantizationType SafeGetQuantizationType(
Dembedding-network.fbs27 // Should be kept in sync with the C++ enum nlp_saft::QuantizationType.
28 enum QuantizationType : byte {
48 quantization_type:QuantizationType = NONE;
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc48 using optimized_ops::depthwise_conv::QuantizationType;
78 typedef std::tuple<DepthwiseConvImplementation, int, QuantizationType, bool,
109 QuantizationType quantization_type = QuantizationType::kNonPerChannelUint8;
119 template <QuantizationType quantization_type>
140 inline void DispatchDepthwiseConvGeneral<QuantizationType::kPerChannelInt8>( in DispatchDepthwiseConvGeneral()
154 template <QuantizationType quantization_type>
358 inline void DispatchDepthwiseConvImpl<QuantizationType::kPerChannelInt8>( in DispatchDepthwiseConvImpl()
362 QuantizationType::kPerChannelInt8>::ExternalType* input_data, in DispatchDepthwiseConvImpl()
365 QuantizationType::kPerChannelInt8>::ExternalType* filter_data, in DispatchDepthwiseConvImpl()
369 QuantizationType::kPerChannelInt8>::ExternalType* output_data) { in DispatchDepthwiseConvImpl()
[all …]
Ddepthwiseconv_per_channel_quantized_test.cc219 optimized_ops::depthwise_conv::QuantizationType::kNonPerChannelUint8>( in GenerateValidShapeConfigurations()
303 optimized_ops::depthwise_conv::QuantizationType::kPerChannelInt8>( in TryTestOneDepthwiseConv3x3Filter()
/external/tensorflow/tensorflow/lite/testing/model_coverage/
Dmodel_coverage_quantization_test.py41 class QuantizationType(enum.Enum): class
47 ALL_QUANTIZATION_TYPES = list(QuantizationType)
98 if quantization_type == QuantizationType.FLOAT16:
100 elif quantization_type in (QuantizationType.FULL_INTEGER,
101 QuantizationType.FULL_INTEGER_16X8):
110 if quantization_type == QuantizationType.FULL_INTEGER_16X8:
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_3x3_filter_common.h224 enum class QuantizationType {
229 template <QuantizationType quantization_type>
233 struct QuantizationTypeImpl<QuantizationType::kNonPerChannelUint8> {
241 struct QuantizationTypeImpl<QuantizationType::kPerChannelInt8> {
249 QuantizationType quantization_type = QuantizationType::kNonPerChannelUint8>
279 if (quantization_type == QuantizationType::kPerChannelInt8) {
438 QuantizationType quantization_type = QuantizationType::kNonPerChannelUint8>
507 QuantizationType quantization_type>
526 QuantizationType quantization_type,
545 QuantizationType quantization_type,
Ddepthwiseconv_uint8_3x3_filter.h5823 QuantizationType::kNonPerChannelUint8> {
5942 QuantizationType::kPerChannelInt8> {
6055 template <QuantizationType quantization_type>
6137 if (quantization_type == QuantizationType::kNonPerChannelUint8) {
6155 if (quantization_type == QuantizationType::kNonPerChannelUint8) {
6174 if (quantization_type == QuantizationType::kNonPerChannelUint8) {
6188 if (quantization_type == QuantizationType::kNonPerChannelUint8) {
6214 if (quantization_type == QuantizationType::kNonPerChannelUint8) {
6249 if (quantization_type == QuantizationType::kNonPerChannelUint8) {
6286 template <QuantizationType quantization_type>
[all …]
Ddepthwiseconv_uint8_transitional.h71 template <QuantizationType quantization_type>
174 template <QuantizationType quantization_type>
290 template <QuantizationType quantization_type>
308 if (quantization_type == QuantizationType::kNonPerChannelUint8) {
369 if (quantization_type == QuantizationType::kNonPerChannelUint8) {
435 template <QuantizationType quantization_type, int32 max_padding>
636 template <QuantizationType quantization_type, int32 max_padding>
755 template <QuantizationType quantization_type>
910 template <QuantizationType quantization_type>
932 template <QuantizationType quantization_type, int32 max_padding>
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h293 optimized_ops::depthwise_conv::QuantizationType::kNonPerChannelUint8>( in DepthwiseConvHybridWithRounding()
Ddepthwise_conv.h1829 optimized_ops::depthwise_conv::QuantizationType::kPerChannelInt8>(
1857 optimized_ops::depthwise_conv::QuantizationType::kPerChannelInt8>(