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

  /external/libtextclassifier/lang_id/common/
embedding-network-params.cc 33 if (s == "FLOAT16") {
34 return QuantizationType::FLOAT16;
embedding-network.cc 43 case QuantizationType::FLOAT16:
44 return cols * sizeof(float16);
121 case QuantizationType::FLOAT16: {
123 // identical, except for float16 (instead of float) and the Float16To32
127 const float16 *weight_ptr =
128 reinterpret_cast<const float16 *>(weights.elements);
embedding-network-params.h 23 #include "lang_id/common/lite_base/float16.h"
38 // lang_id/common/float16.h
39 FLOAT16,
94 const ::libtextclassifier3::mobile::float16 *quant_scales = nullptr;
216 virtual const ::libtextclassifier3::mobile::float16 *embeddings_quant_scales(
  /external/libtextclassifier/lang_id/common/flatbuffers/
embedding-network-params-from-flatbuffer.cc 61 matrix.rows * sizeof(float16), bytes);
107 case QuantizationType::FLOAT16: {
108 element_range_size_in_bytes = num_elements * sizeof(float16);
110 // No need to verify the scales: FLOAT16 quantization does not use scales.
381 const float16 *EmbeddingNetworkParamsFromFlatbuffer::SafeGetScales(
418 return QuantizationType::FLOAT16;

Completed in 138 milliseconds