HomeSort by relevance Sort by last modified time
    Searched refs:QuantizationType (Results 1 - 5 of 5) sorted by null

  /external/libtextclassifier/lang_id/common/
embedding-network-params.cc 23 QuantizationType ParseQuantizationType(const string &s) {
25 return QuantizationType::NONE;
28 return QuantizationType::UINT8;
31 return QuantizationType::UINT4;
34 return QuantizationType::FLOAT16;
41 return QuantizationType::NONE;
embedding-network-params.h 28 enum class QuantizationType {
46 // Converts "UINT8" -> QuantizationType::UINT8, and so on.
47 QuantizationType ParseQuantizationType(const string &s);
87 QuantizationType quant_type = QuantizationType::NONE;
138 matrix.quant_type = QuantizationType::NONE;
174 matrix.quant_type = QuantizationType::NONE;
212 virtual QuantizationType embeddings_quant_type(int i) const {
213 return QuantizationType::NONE;
233 virtual QuantizationType hidden_weights_quant_type(int i) const
    [all...]
embedding-network.cc 27 SAFTM_CHECK_EQ(static_cast<int>(QuantizationType::NONE),
34 QuantizationType quant_type = matrix.quant_type;
36 case QuantizationType::NONE:
38 case QuantizationType::UINT8:
40 case QuantizationType::UINT4:
43 case QuantizationType::FLOAT16:
93 case QuantizationType::NONE: {
121 case QuantizationType::FLOAT16: {
122 // See comments for the QuantizationType::NONE case: the code is almost
198 case QuantizationType::NONE:
    [all...]
  /external/libtextclassifier/lang_id/common/flatbuffers/
embedding-network-params-from-flatbuffer.cc 86 case QuantizationType::NONE:
89 case QuantizationType::UINT8: {
96 case QuantizationType::UINT4: {
107 case QuantizationType::FLOAT16: {
401 QuantizationType EmbeddingNetworkParamsFromFlatbuffer::SafeGetQuantizationType(
404 return QuantizationType::NONE;
406 saft_fbs::QuantizationType quantization_type = matrix->quantization_type();
408 // Conversion from nlp_saft::saft_fbs::QuantizationType to
409 // nlp_saft::QuantizationType (due to legacy reasons, we have both).
412 return QuantizationType::NONE
    [all...]
embedding-network-params-from-flatbuffer.h 77 QuantizationType embeddings_quant_type(int i) const override {
102 QuantizationType hidden_weights_quant_type(int i) const override {
141 QuantizationType softmax_weights_quant_type(int i) const override {
259 QuantizationType SafeGetQuantizationType(

Completed in 77 milliseconds