Home
last modified time | relevance | path

Searched defs:packed_weight (Results 1 – 25 of 25) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A Dqconv_unpack.cpp53 const c10::intrusive_ptr<ConvPackedParamsBase<kSpatialDim>>& packed_weight) { in run()
89 const c10::intrusive_ptr<ConvPackedParamsBase<2>>& packed_weight) { in run()
131 const c10::intrusive_ptr<ConvPackedParamsBase<kSpatialDim>>& packed_weight) { in run()
140 const c10::intrusive_ptr<ConvPackedParamsBase<kSpatialDim>>& packed_weight) { in run()
149 const c10::intrusive_ptr<ConvPackedParamsBase<kSpatialDim>>& packed_weight) { in run()
158 const c10::intrusive_ptr<ConvPackedParamsBase<kSpatialDim>>& packed_weight) { in run()
167 const c10::intrusive_ptr<ConvPackedParamsBase<kSpatialDim>>& packed_weight) { in run()
176 const c10::intrusive_ptr<ConvPackedParamsBase<kSpatialDim>>& packed_weight) { in run()
H A Dqlinear_unpack.cpp25 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight) { in run()
33 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight) { in run()
48 const at::Tensor& packed_weight) { in run()
59 const at::Tensor& packed_weight) { in run()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqembeddingbag_unpack.cpp23 auto packed_weight = packed_w; in unpack() local
106 Tensor& qembeddingbag_byte_unpack_out(Tensor& output, const Tensor& packed_weight) { in qembeddingbag_byte_unpack_out()
161 Tensor qembeddingbag_byte_unpack(const Tensor& packed_weight) { in qembeddingbag_byte_unpack()
170 Tensor qembeddingbag_byte_unpack_meta(const Tensor& packed_weight) { in qembeddingbag_byte_unpack_meta()
186 const Tensor& packed_weight, in _qembeddingbag_nbit_unpack_helper()
245 Tensor qembeddingbag_4bit_unpack(const Tensor& packed_weight) { in qembeddingbag_4bit_unpack()
257 Tensor qembeddingbag_2bit_unpack(const Tensor& packed_weight) { in qembeddingbag_2bit_unpack()
264 const c10::intrusive_ptr<EmbeddingPackedParamsBase>& packed_weight) { in run()
H A Dqlinear.cpp986 auto packed_weight = at::native::itensor_from_mkldnn(onednn_weight); in linear_int8_with_onednn_weight() local
1113 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight, in run()
1130 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight, in run()
1155 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight, in run()
1181 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight) { in run()
H A Dqconv_dynamic.cpp193 const c10::intrusive_ptr<ConvPackedParamsBase<kSpatialDim>>& in run()
205 const c10::intrusive_ptr<ConvPackedParamsBase<2>>& packed_weight, in run()
H A Dqlinear_dynamic.cpp630 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight, in run()
646 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight) { in run()
692 auto packed_weight = PackedLinearWeightFp16::prepack(weight, bias); in run() local
H A Dqconv.cpp1551 auto packed_weight = at::native::itensor_from_mkldnn(weight); in _quantized_convolution_onednn() local
1776 const c10::intrusive_ptr<ConvPackedParamsBase<kSpatialDim>>& packed_weight, in run()
1793 const c10::intrusive_ptr<ConvPackedParamsBase<kSpatialDim>>& packed_weight, in run()
1822 const c10::intrusive_ptr<ConvPackedParamsBase<2>>& packed_weight, in run()
1844 const c10::intrusive_ptr<ConvPackedParamsBase<kSpatialDim>>& packed_weight, in run()
H A Dqlinear_prepack.cpp305 auto packed_weight = at::native::new_with_itensor_mkldnn( in pack_weight_to_onednn_tensor() local
486 const at::Tensor& packed_weight, in _wrapped_quantized_linear_prepacked()
544 [[maybe_unused]] const at::Tensor& packed_weight, in _wrapped_quantized_linear_prepacked_meta()
H A Dqembeddingbag.cpp1039 const c10::intrusive_ptr<EmbeddingPackedParamsBase>& packed_weight, in run()
1078 const c10::intrusive_ptr<EmbeddingPackedParamsBase>& packed_weight, in run()
H A Dqconv_prepack.cpp608 auto packed_weight = at::native::new_with_itensor_mkldnn( in _qconv_prepack_onednn() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DQuantizedLinear.cpp406 c10::intrusive_ptr<LinearPackedParamsBase> packed_weight = in fbgemm_pack_gemm_matrix_fp16() local
416 const Tensor& packed_weight, in fbgemm_linear_fp16_weight_fp32_activation()
464 const Tensor& packed_weight, in fbgemm_linear_fp16_weight()
H A DRNN.cpp344 at::Tensor packed_weight = in make_quantized_cell_params() local
/aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/
H A Dqembedding_pack_test.py73 def forward(self, packed_weight): argument
85 def forward(self, packed_weight): argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/
H A DConv.cpp354 const c10::intrusive_ptr<ConvPackedParamsBase<2>>& packed_weight, in run()
377 const c10::intrusive_ptr<ConvPackedParamsBase<kSpatialDim>>& packed_weight, in run()
H A DLinear.cpp347 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight, in run()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cuda/
H A DEmbeddingBag.cu172 Tensor qembeddingbag_byte_unpack(const Tensor& packed_weight) { in qembeddingbag_byte_unpack()
544 Tensor qembeddingbag_4bit_unpack(const Tensor& packed_weight) { in qembeddingbag_4bit_unpack()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/
H A Dqlinear_unpack.cpp130 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight) { in run()
H A Dqlinear_dynamic.cpp170 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight) { in run()
H A Dqlinear.cpp238 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight, in run()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dops.cpp2221 c10::intrusive_ptr<LinearPackedParamsBase> packed_weight; in __anon11f46a8b6e02() local
2266 c10::intrusive_ptr<LinearPackedParamsBase> packed_weight; in __anon11f46a8b7002() local
2314 c10::intrusive_ptr<LinearPackedParamsBase> packed_weight, in apply_dynamic_out_functor()
2323 c10::intrusive_ptr<LinearPackedParamsBase> packed_weight, in apply_dynamic_out_functor()
2337 c10::intrusive_ptr<LinearPackedParamsBase> packed_weight; in quantized_linear_dynamic_fp16_impl() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dmkldnn_rewrite.cpp178 Value* packed_weight = graph->insertConstant(weak_class_obj) in PrePackingOpsFolder() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DConvPrepack.cpp83 ideep::tensor packed_weight; in create() local
H A DMKLDNNConversions.cpp609 auto packed_weight = empty_mkldnn( in mkl_reorder_linear_weight() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dunpack_quantized_weights.cpp475 at::Tensor packed_weight = itr->second.toTensor(); in unpackQuantizedWeightsHelper() local
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py2315 def meta_mkl_linear(input_tensor, packed_weight, orig_weight, bias, batch_size): argument