Home
last modified time | relevance | path

Searched refs:Quantizer (Results 1 – 25 of 75) sorted by relevance

123

/external/pytorch/aten/src/ATen/core/
DQuantizerBase.h11 struct Quantizer;
12 using ConstQuantizerPtr = const c10::intrusive_ptr<Quantizer>&;
13 using QuantizerPtr = c10::intrusive_ptr<Quantizer>;
39 struct TORCH_API Quantizer : public c10::intrusive_ptr_target { struct
42 explicit Quantizer(ScalarType scalar_type) : scalar_type_(scalar_type) {} in Quantizer() argument
43 ~Quantizer() override;
51 return c10::intrusive_ptr<Quantizer>::reclaim(this); in intrusive_from_this()
/external/icing/icing/index/embed/
Dquantizer.h33 class Quantizer {
42 static libtextclassifier3::StatusOr<Quantizer> Create(float float_min, in Create()
53 return Quantizer(float_min, scale_factor); in Create()
57 explicit Quantizer(const char* data) { in Quantizer() function
58 memcpy(this, data, sizeof(Quantizer)); in Quantizer()
82 explicit Quantizer(float float_min, float scale_factor) in Quantizer() function
Dquantizer_test.cc41 EXPECT_THAT(Quantizer::Create(/*float_min=*/1.0f, /*float_max=*/0.0f), in TEST()
50 ICING_ASSERT_OK_AND_ASSIGN(Quantizer quantizer, in TEST()
51 Quantizer::Create(float_min, float_max)); in TEST()
91 ICING_ASSERT_OK_AND_ASSIGN(Quantizer quantizer, in TEST()
92 Quantizer::Create(float_min, float_max)); in TEST()
131 Quantizer quantizer, in TEST()
132 Quantizer::Create(/*float_min=*/0.0f, /*float_max=*/255.0f)); in TEST()
149 ICING_ASSERT_OK_AND_ASSIGN(Quantizer quantizer, in TEST()
150 Quantizer::Create(/*float_min=*/constant_value, in TEST()
Dembedding-scorer_test.cc31 const Quantizer& quantizer) { in QuantizeVector()
48 Quantizer quantizer, in TEST()
49 Quantizer::Create(/*float_min=*/-1.0f, /*float_max=*/1.0f)); in TEST()
76 Quantizer quantizer, in TEST()
77 Quantizer::Create(/*float_min=*/-1.0f, /*float_max=*/1.0f)); in TEST()
104 Quantizer quantizer, in TEST()
105 Quantizer::Create(/*float_min=*/-1.0f, /*float_max=*/1.0f)); in TEST()
Dembedding-scorer.h36 const Quantizer& quantizer) const = 0;
45 const Quantizer& quantizer) const override;
52 const Quantizer& quantizer) const override;
59 const Quantizer& quantizer) const override;
Dembedding-scorer.cc41 T value, const Quantizer&) { in ToFloat() argument
47 T quantized, const Quantizer& quantizer) { in ToFloat()
123 const Quantizer& quantizer) const { in Score()
129 const Quantizer& quantizer) const { in Score()
135 const Quantizer& quantizer) const { in Score()
Dembedding-index.cc117 libtextclassifier3::StatusOr<Quantizer> CreateQuantizer( in CreateQuantizer()
124 return Quantizer::Create(*minmax_pair.first, *minmax_pair.second); in CreateQuantizer()
271 ICING_ASSIGN_OR_RETURN(Quantizer quantizer, CreateQuantizer(vector)); in AppendEmbeddingVector()
283 quantized_embedding_vectors_->Allocate(sizeof(Quantizer) + dimension)); in AppendEmbeddingVector()
285 sizeof(Quantizer)); in AppendEmbeddingVector()
286 mutable_arr.SetArray(/*idx=*/sizeof(Quantizer), in AppendEmbeddingVector()
397 sizeof(Quantizer) + dimension)); in TransferEmbeddingVector()
398 mutable_arr.SetArray(/*idx=*/0, old_data, sizeof(Quantizer) + dimension); in TransferEmbeddingVector()
585 Quantizer quantizer(data); in ScoreEmbeddingHit()
587 reinterpret_cast<const uint8_t*>(data + sizeof(Quantizer)); in ScoreEmbeddingHit()
Dembedding-index_test.cc380 Eq(3 + sizeof(Quantizer))); in TEST_F()
431 /*location=*/3 + sizeof(Quantizer)); in TEST_F()
436 Eq(2 * (3 + sizeof(Quantizer)))); // Two quantized vectors in TEST_F()
595 Eq(3 + sizeof(Quantizer))); in TEST_F()
650 Eq(3 + sizeof(Quantizer))); in TEST_F()
869 Eq(3 + sizeof(Quantizer))); in TEST_F()
887 Eq(3 + sizeof(Quantizer))); in TEST_F()
907 Eq(3 + sizeof(Quantizer))); in TEST_F()
963 Eq(3 + sizeof(Quantizer))); in TEST_F()
988 Eq(3 + sizeof(Quantizer))); in TEST_F()
[all …]
/external/pytorch/torch/ao/quantization/quantizer/
Dcomposable_quantizer.py5 from .quantizer import QuantizationAnnotation, Quantizer
17 class ComposableQuantizer(Quantizer):
37 def __init__(self, quantizers: List[Quantizer]): argument
43 self, gm: torch.fx.GraphModule, quantizer: Quantizer argument
Dembedding_quantizer.py13 Quantizer,
44 class EmbeddingQuantizer(Quantizer):
D__init__.py8 Quantizer,
/external/pytorch/aten/src/ATen/quantized/
DQuantizer.h28 struct TORCH_API UnknownQuantizer : public Quantizer {
30 : Quantizer(scalar_type) {} in UnknownQuantizer()
45 struct TORCH_API UniformQuantizer : public Quantizer {
46 explicit UniformQuantizer(ScalarType scalar_type) : Quantizer(scalar_type) {} in UniformQuantizer()
54 struct TORCH_API NonUniformQuantizer : public Quantizer {
55 explicit NonUniformQuantizer(ScalarType scalar_type) : Quantizer(scalar_type) {} in NonUniformQuantizer()
/external/executorch/docs/source/
Dquantization-overview.md10 …model should be quantized, expressed in a backend specific ``Quantizer`` class. ``Quantizer`` prov…
12Quantizer`` to express how different operators or operator patterns are quantized in their backend…
14 Modeling users will use the ``Quantizer`` specific to their target backend to quantize their model,…
/external/pytorch/tools/autograd/templates/
DVariableType.h20 struct Quantizer;
40 using at::Quantizer;
44 using ConstQuantizerPtr = const c10::intrusive_ptr<Quantizer>&;
/external/pytorch/aten/src/ATen/
DTypeDefault.h24 struct Quantizer;
28 using ConstQuantizerPtr = const c10::intrusive_ptr<Quantizer>&;
/external/libgav1/src/
Dquantizer.cc326 Quantizer::Quantizer(int bitdepth, const QuantizerParameters* params) in Quantizer() function in libgav1::Quantizer
334 int Quantizer::GetDcValue(Plane plane, int qindex) const { in GetDcValue()
339 int Quantizer::GetAcValue(Plane plane, int qindex) const { in GetAcValue()
Dquantizer_test.cc63 Quantizer quantizer(8, &params); in TEST()
88 Quantizer quantizer(10, &params); in TEST()
114 Quantizer quantizer(12, &params); in TEST()
145 Quantizer quantizer(8, &params); in TEST()
170 Quantizer quantizer(10, &params); in TEST()
196 Quantizer quantizer(12, &params); in TEST()
Dquantizer.h36 class Quantizer {
38 Quantizer(int bitdepth, const QuantizerParameters* params);
/external/pytorch/test/quantization/pt2e/
Dtest_metadata_porting.py9 from torch.ao.quantization.quantizer import QuantizationAnnotation, Quantizer
149 class BackendAQuantizer(Quantizer):
216 class BackendAQuantizer(Quantizer):
260 class BackendAQuantizer(Quantizer):
330 class BackendAQuantizer(Quantizer):
394 class BackendAQuantizer(Quantizer):
430 class BackendAQuantizer(Quantizer):
475 class BackendAQuantizer(Quantizer):
Dtest_duplicate_dq.py17 Quantizer,
128 class BackendAQuantizer(Quantizer):
164 class BackendAQuantizer(Quantizer):
201 class BackendAQuantizer(Quantizer):
283 class BackendAQuantizer(Quantizer):
/external/pytorch/torch/ao/quantization/
Dquantize_pt2e.py11 Quantizer,
33 quantizer: Quantizer, argument
110 quantizer: Quantizer, argument
/external/executorch/backends/mediatek/quantizer/
Dquantizer.py7 from torch.ao.quantization.quantizer import Quantizer
17 class NeuropilotQuantizer(Quantizer):
/external/icing/icing/testing/
Dembedding-test-utils.cc75 Quantizer quantizer(data); in GetAndRestoreQuantizedEmbeddingVectorFromIndex()
77 reinterpret_cast<const uint8_t*>(data + sizeof(Quantizer)); in GetAndRestoreQuantizedEmbeddingVectorFromIndex()
/external/executorch/backends/example/
Dexample_quantizer.py15 from torch.ao.quantization.quantizer import QuantizationSpec, Quantizer
61 class ExampleQuantizer(Quantizer):
/external/executorch/backends/vulkan/quantizer/
Dvulkan_quantizer.py17 from torch.ao.quantization.quantizer import QuantizationSpec, Quantizer
75 class VulkanQuantizer(Quantizer):

123