Home
last modified time | relevance | path

Searched refs:quant_type (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dfill_test.cc76 template <typename dims_type, typename quant_type>
93 QuantizeAndPopulate<quant_type>(value_, {value}); in QuantizedFillOpModel()
96 std::vector<quant_type> GetOutput() { in GetOutput()
97 return ExtractVector<quant_type>(output_); in GetOutput()
182 template <typename quant_type>
188 std::numeric_limits<quant_type>::max() / in QuantizedFill()
189 static_cast<float>(std::numeric_limits<quant_type>::max() + 1); in QuantizedFill()
190 const TensorData tensor_data(GetTensorType<quant_type>(), {}, in QuantizedFill()
193 QuantizedFillOpModel<int32_t, quant_type> m(TensorType_INT32, {2}, {2, 3}, in QuantizedFill()
201 (max_value - min_value) / (std::numeric_limits<quant_type>::max() - in QuantizedFill()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_quantize_helper.h293 UniformQuantizedType quant_type = nullptr; in processConstantOp() local
306 quant_type = UniformQuantizedType::getChecked( in processConstantOp()
311 quant_type = in processConstantOp()
318 if (!quant_type) { in processConstantOp()
325 Type cast_type = quant_type.castFromExpressedType(expressed_type); in processConstantOp()
350 quant::QuantizedType quant_type; in replaceStatsOp() local
368 quant_type = UniformQuantizedType::getChecked( in replaceStatsOp()
379 quant_type = quant::fakeQuantAttrsToType( in replaceStatsOp()
384 quant_type = quant::fakeQuantAttrsToType( in replaceStatsOp()
390 quant_type = quant::DownCastScale(quant_type, min, max, op.getLoc()); in replaceStatsOp()
[all …]
/external/libtextclassifier/native/lang_id/common/
Dembedding-network.cc28 static_cast<int>(matrix.quant_type)) in CheckNoQuantization()
34 QuantizationType quant_type = matrix.quant_type; in GetMatrixRowSizeInBytes() local
35 switch (quant_type) { in GetMatrixRowSizeInBytes()
47 << static_cast<int>(quant_type); in GetMatrixRowSizeInBytes()
92 switch (weights.quant_type) { in SparseReluProductPlusBias()
143 << static_cast<int>(weights.quant_type); in SparseReluProductPlusBias()
197 switch (embedding_matrix.quant_type) { in ConcatEmbeddings()
237 << static_cast<int>(embedding_matrix.quant_type); in ConcatEmbeddings()
Dembedding-network-params.h87 QuantizationType quant_type = QuantizationType::NONE; member
106 matrix.quant_type = embeddings_quant_type(i); in GetEmbeddingMatrix()
122 matrix.quant_type = hidden_weights_quant_type(i); in GetHiddenLayerMatrix()
138 matrix.quant_type = QuantizationType::NONE; in GetHiddenLayerBias()
159 matrix.quant_type = softmax_weights_quant_type(0); in GetSoftmaxMatrix()
174 matrix.quant_type = QuantizationType::NONE; in GetSoftmaxBias()
/external/tensorflow/tensorflow/lite/testing/model_coverage/
Dmodel_coverage_quantization_test.py65 (quant_type.value, quant_type) for quant_type in quantization_types)
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_utils.h107 quant::QuantizedType quant_type; in matchAndRewrite() local
128 quant_type = in matchAndRewrite()
132 quant_type = DownCastScale(quant_type, mins, maxs, op->getLoc()); in matchAndRewrite()
143 quant_type = in matchAndRewrite()
147 quant_type = DownCastScale(quant_type, rmin, rmax, op->getLoc()); in matchAndRewrite()
154 Type result_type = quant_type.castFromExpressedType(op.getType()); in matchAndRewrite()
Dquantization_driver.cc454 if (auto quant_type = final_type.dyn_cast_or_null<quant::QuantizedType>()) { in SetConstantResultParams() local
455 return SetResultParams(op, 0, quant_type); in SetConstantResultParams()
/external/tensorflow/tensorflow/lite/python/
Dutil.py684 quant_type = _convert_tflite_enum_type_to_tf_type(quant_tensor.type)
685 if quant_type not in _MAP_QUANT_TO_IO_TYPES:
692 get_tf_type_name(quant_type)))
694 inference_io_types = _MAP_QUANT_TO_IO_TYPES[quant_type]
778 quant_type = _convert_tflite_enum_type_to_tf_type(quant_tensor.type)
779 if quant_type not in _MAP_QUANT_TO_IO_TYPES:
786 get_tf_type_name(quant_type)))
788 inference_io_types = _MAP_QUANT_TO_IO_TYPES[quant_type]
/external/libtextclassifier/native/lang_id/common/flatbuffers/
Dembedding-network-params-from-flatbuffer.cc56 << static_cast<int>(matrix.quant_type) in VerifyQuantizationScales()
85 switch (matrix.quant_type) { in VerifyMatrix()
115 << static_cast<int>(matrix.quant_type); in VerifyMatrix()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_weights.h48 BufferType quant_type = BufferType::QUANTIZED_INT8);
Dquantize_weights.cc650 const Model* input_model, BufferType quant_type) { in QuantizeWeights() argument
651 switch (quant_type) { in QuantizeWeights()
/external/mesa3d/src/gallium/frontends/va/
Dpicture_mpeg4.c53 context->desc.mpeg4.quant_type = mpeg4->vol_fields.bits.quant_type; in vlVaHandlePictureParameterBufferMPEG4()
/external/mesa3d/src/gallium/drivers/nouveau/
Dnouveau_vp3_video_vp.c71 uint8_t quant_type; // bool, written to 528 member
295 pic_vp->quant_type = desc->quant_type; in nouveau_vp3_fill_picparm_mpeg4_vp()
/external/mesa3d/src/gallium/drivers/r600/
Dradeon_uvd.h313 uint8_t quant_type; member
Dradeon_uvd.c640 result.quant_type = pic->quant_type; in get_mpeg4_msg()
/external/mesa3d/src/gallium/drivers/radeon/
Dradeon_uvd.h313 uint8_t quant_type; member
Dradeon_vcn_dec.h525 unsigned char quant_type; member
Dradeon_uvd.c947 result.quant_type = pic->quant_type; in get_mpeg4_msg()
Dradeon_vcn_dec.c802 result.quant_type = pic->quant_type; in get_mpeg4_msg()
/external/mesa3d/src/gallium/frontends/vdpau/
Ddecode.c293 picture->quant_type = picture_info->quant_type; in vlVdpDecoderRenderMpeg4()
/external/mesa3d/src/gallium/include/pipe/
Dp_video_state.h244 uint8_t quant_type; member