Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/libtextclassifier/native/lang_id/common/
H A 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 …]
H A Dembedding-network.cc29 SAFTM_CHECK_EQ(static_cast<int>(QuantizationType::NONE), in CheckNoQuantization()
36 QuantizationType quant_type = matrix.quant_type; in GetMatrixRowSizeInBytes()
38 case QuantizationType::NONE: in GetMatrixRowSizeInBytes()
40 case QuantizationType::UINT8: in GetMatrixRowSizeInBytes()
42 case QuantizationType::UINT4: in GetMatrixRowSizeInBytes()
45 case QuantizationType::FLOAT16: in GetMatrixRowSizeInBytes()
95 case QuantizationType::NONE: { in SparseReluProductPlusBias()
123 case QuantizationType::FLOAT16: { in SparseReluProductPlusBias()
201 case QuantizationType::NONE: { in ConcatEmbeddings()
209 case QuantizationType::UINT8: { in ConcatEmbeddings()
[all …]
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A 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 …]
H A Ddepthwiseconv_per_channel_quantized_test.cc219 optimized_ops::depthwise_conv::QuantizationType::kNonPerChannelUint8>( in GenerateValidShapeConfigurations()
303 optimized_ops::depthwise_conv::QuantizationType::kPerChannelInt8>( in TryTestOneDepthwiseConv3x3Filter()
/aosp_15_r20/external/libtextclassifier/native/lang_id/common/flatbuffers/
H A 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 …]
H A 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(
H A Dembedding-network.fbs27 // Should be kept in sync with the C++ enum nlp_saft::QuantizationType.
28 enum QuantizationType : byte {
48 quantization_type:QuantizationType = NONE;
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A 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,
H A Ddepthwiseconv_uint8_3x3_filter.h5824 QuantizationType::kNonPerChannelUint8> {
5943 QuantizationType::kPerChannelInt8> {
6056 template <QuantizationType quantization_type>
6138 if (quantization_type == QuantizationType::kNonPerChannelUint8) {
6156 if (quantization_type == QuantizationType::kNonPerChannelUint8) {
6175 if (quantization_type == QuantizationType::kNonPerChannelUint8) {
6189 if (quantization_type == QuantizationType::kNonPerChannelUint8) {
6215 if (quantization_type == QuantizationType::kNonPerChannelUint8) {
6250 if (quantization_type == QuantizationType::kNonPerChannelUint8) {
6287 template <QuantizationType quantization_type>
[all …]
H A Ddepthwiseconv_uint8_transitional.h88 template <QuantizationType quantization_type>
191 template <QuantizationType quantization_type>
307 template <QuantizationType quantization_type>
325 if (quantization_type == QuantizationType::kNonPerChannelUint8) {
386 if (quantization_type == QuantizationType::kNonPerChannelUint8) {
452 template <QuantizationType quantization_type, int32 max_padding>
653 template <QuantizationType quantization_type, int32 max_padding>
772 template <QuantizationType quantization_type>
927 template <QuantizationType quantization_type>
949 template <QuantizationType quantization_type, int32 max_padding>
[all …]
/aosp_15_r20/external/icing/icing/
H A Dschema-builder.h66 constexpr EmbeddingIndexingConfig::QuantizationType::Code
67 QUANTIZATION_TYPE_NONE = EmbeddingIndexingConfig::QuantizationType::NONE;
68 constexpr EmbeddingIndexingConfig::QuantizationType::Code
70 EmbeddingIndexingConfig::QuantizationType::QUANTIZE_8_BIT;
182 EmbeddingIndexingConfig::QuantizationType::Code quantization_type =
183 EmbeddingIndexingConfig::QuantizationType::NONE) {
/aosp_15_r20/external/icing/icing/index/embed/
H A Dembedding-index.cc255 EmbeddingIndexingConfig::QuantizationType::Code quantization_type) { in AppendEmbeddingVector()
259 quantization_type == EmbeddingIndexingConfig::QuantizationType::NONE) { in AppendEmbeddingVector()
290 EmbeddingIndexingConfig::QuantizationType::Code quantization_type) { in BufferEmbedding()
370 EmbeddingIndexingConfig::QuantizationType::Code quantization_type, in TransferEmbeddingVector()
374 quantization_type == EmbeddingIndexingConfig::QuantizationType::NONE) { in TransferEmbeddingVector()
465 EmbeddingIndexingConfig::QuantizationType::Code quantization_type, in TransferIndex()
567 EmbeddingIndexingConfig::QuantizationType::Code quantization_type) const { in ScoreEmbeddingHit()
571 quantization_type == EmbeddingIndexingConfig::QuantizationType::NONE) { in ScoreEmbeddingHit()
H A Dembedding-index.h124 EmbeddingIndexingConfig::QuantizationType::Code quantization_type);
210 EmbeddingIndexingConfig::QuantizationType::Code quantization_type) const;
285 EmbeddingIndexingConfig::QuantizationType::Code quantization_type,
329 EmbeddingIndexingConfig::QuantizationType::Code quantization_type);
H A Ddoc-hit-info-iterator-embedding.cc145 EmbeddingIndexingConfig::QuantizationType::Code quantization_type, in AdvanceToNextUnfilteredDocument()
/aosp_15_r20/packages/modules/AppSearch/service/java/com/android/server/appsearch/external/localstorage/converter/
DSchemaToProtoConverter.java318 EmbeddingIndexingConfig.QuantizationType.Code embeddingQuantizationType = in toEmbeddingPropertyConfig()
488 private static EmbeddingIndexingConfig.QuantizationType.Code
490 @AppSearchSchema.EmbeddingPropertyConfig.QuantizationType in convertEmbeddingQuantizationTypeToProto()
494 return EmbeddingIndexingConfig.QuantizationType.Code.NONE; in convertEmbeddingQuantizationTypeToProto()
496 return EmbeddingIndexingConfig.QuantizationType.Code.QUANTIZE_8_BIT; in convertEmbeddingQuantizationTypeToProto()
502 @AppSearchSchema.EmbeddingPropertyConfig.QuantizationType
504 @NonNull EmbeddingIndexingConfig.QuantizationType.Code quantizationType) { in convertEmbeddingQuantizationTypeTypeFromProto()
/aosp_15_r20/out/soong/.intermediates/external/libtextclassifier/native/libtextclassifier_fbgen_lang_id_common_flatbuffers_embedding-network/gen/lang_id/common/flatbuffers/
Dembedding-network_generated.h35 enum QuantizationType : int8_t { enum
44 inline const QuantizationType (&EnumValuesQuantizationType())[4] { in EnumValuesQuantizationType()
45 static const QuantizationType values[] = { in EnumValuesQuantizationType()
65 inline const char *EnumNameQuantizationType(QuantizationType e) { in EnumNameQuantizationType()
75 …libtextclassifier3::saft_fbs::QuantizationType quantization_type = libtextclassifier3::saft_fbs::Q…
98 libtextclassifier3::saft_fbs::QuantizationType quantization_type() const { in quantization_type()
99 …return static_cast<libtextclassifier3::saft_fbs::QuantizationType>(GetField<int8_t>(VT_QUANTIZATIO… in quantization_type()
138 void add_quantization_type(libtextclassifier3::saft_fbs::QuantizationType quantization_type) { in add_quantization_type()
165 …libtextclassifier3::saft_fbs::QuantizationType quantization_type = libtextclassifier3::saft_fbs::Q…
183 …libtextclassifier3::saft_fbs::QuantizationType quantization_type = libtextclassifier3::saft_fbs::Q…
/aosp_15_r20/packages/modules/AppSearch/framework/java/external/android/app/appsearch/safeparcel/
DPropertyConfigParcel.java254 @AppSearchSchema.EmbeddingPropertyConfig.QuantizationType int quantizationType) { in createForEmbedding()
703 @AppSearchSchema.EmbeddingPropertyConfig.QuantizationType
712 @Param(id = 2) @AppSearchSchema.EmbeddingPropertyConfig.QuantizationType in EmbeddingIndexingConfigParcel()
725 @AppSearchSchema.EmbeddingPropertyConfig.QuantizationType
/aosp_15_r20/external/icing/icing/schema/
H A Dsection.h110 EmbeddingIndexingConfig::QuantizationType::Code quantization_type;
119 EmbeddingIndexingConfig::QuantizationType::Code quantization_type_in, in SectionMetadata()
H A Dschema-store.h476 libtextclassifier3::StatusOr<EmbeddingIndexingConfig::QuantizationType::Code>
/aosp_15_r20/packages/modules/AppSearch/testing/servicestests/src/com/android/server/appsearch/external/localstorage/converter/
DSchemaToProtoConverterTest.java169 .QuantizationType.Code in testGetProto_DescriptionSet()
580 .QuantizationType.Code in testGetProto_EmbeddingProperty()
597 .QuantizationType.Code in testGetProto_EmbeddingProperty()
/aosp_15_r20/packages/modules/AppSearch/framework/java/external/android/app/appsearch/
DAppSearchSchema.java1854 public @interface QuantizationType {} annotation in AppSearchSchema.EmbeddingPropertyConfig
1882 @EmbeddingPropertyConfig.QuantizationType
1901 @EmbeddingPropertyConfig.QuantizationType
1972 @EmbeddingPropertyConfig.QuantizationType int quantizationType) { in setQuantizationType()
/aosp_15_r20/external/icing/proto/icing/proto/
H A Dschema.proto232 message QuantizationType { message
240 optional QuantizationType.Code quantization_type = 2;
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A Ddepthwise_conv_hybrid.h295 optimized_ops::depthwise_conv::QuantizationType::kNonPerChannelUint8>( in DepthwiseConvHybridWithRounding()
H A Ddepthwise_conv.h1839 optimized_ops::depthwise_conv::QuantizationType::kPerChannelInt8>(
1867 optimized_ops::depthwise_conv::QuantizationType::kPerChannelInt8>(

12