Home
last modified time | relevance | path

Searched refs:QuantizationType (Results 1 – 21 of 21) 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.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 …]
/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/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.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 …]
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 …]
/external/icing/icing/
Dschema-builder.h65 constexpr EmbeddingIndexingConfig::QuantizationType::Code
66 QUANTIZATION_TYPE_NONE = EmbeddingIndexingConfig::QuantizationType::NONE;
67 constexpr EmbeddingIndexingConfig::QuantizationType::Code
69 EmbeddingIndexingConfig::QuantizationType::QUANTIZE_8_BIT;
181 EmbeddingIndexingConfig::QuantizationType::Code quantization_type =
182 EmbeddingIndexingConfig::QuantizationType::NONE) {
Dicing-search-engine_initialization_test.cc6669 EmbeddingIndexingConfig::QuantizationType::NONE) in TEST_P()
6676 QuantizationType::QUANTIZE_8_BIT) in TEST_P()
/external/icing/icing/index/embed/
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);
Dembedding-index.cc260 EmbeddingIndexingConfig::QuantizationType::Code quantization_type) { in AppendEmbeddingVector()
264 quantization_type == EmbeddingIndexingConfig::QuantizationType::NONE) { in AppendEmbeddingVector()
295 EmbeddingIndexingConfig::QuantizationType::Code quantization_type) { in BufferEmbedding()
375 EmbeddingIndexingConfig::QuantizationType::Code quantization_type, in TransferEmbeddingVector()
379 quantization_type == EmbeddingIndexingConfig::QuantizationType::NONE) { in TransferEmbeddingVector()
470 EmbeddingIndexingConfig::QuantizationType::Code quantization_type, in TransferIndex()
572 EmbeddingIndexingConfig::QuantizationType::Code quantization_type) const { in ScoreEmbeddingHit()
576 quantization_type == EmbeddingIndexingConfig::QuantizationType::NONE) { in ScoreEmbeddingHit()
Ddoc-hit-info-iterator-embedding.cc127 EmbeddingIndexingConfig::QuantizationType::Code quantization_type = in AdvanceToNextUnfilteredDocument()
128 EmbeddingIndexingConfig::QuantizationType::NONE; in AdvanceToNextUnfilteredDocument()
/external/icing/icing/schema/
Dsection.h110 EmbeddingIndexingConfig::QuantizationType::Code quantization_type;
119 EmbeddingIndexingConfig::QuantizationType::Code quantization_type_in, in SectionMetadata()
Dschema-store.h489 libtextclassifier3::StatusOr<EmbeddingIndexingConfig::QuantizationType::Code>
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h295 optimized_ops::depthwise_conv::QuantizationType::kNonPerChannelUint8>( in DepthwiseConvHybridWithRounding()
Ddepthwise_conv.h1839 optimized_ops::depthwise_conv::QuantizationType::kPerChannelInt8>(
1867 optimized_ops::depthwise_conv::QuantizationType::kPerChannelInt8>(
/external/icing/proto/icing/proto/
Dschema.proto232 message QuantizationType { message
240 optional QuantizationType.Code quantization_type = 2;