Home
last modified time | relevance | path

Searched defs:LinearPackedParamsBase (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A DPackedParams.h6 struct LinearPackedParamsBase : public torch::jit::CustomClassHolder { struct
17 virtual at::Tensor& apply_out( in apply_out()
28 virtual at::Tensor& apply_relu_out( in apply_relu_out()
54 virtual at::Tensor apply_with_input_q_dq_qweight_dq_output_fp32( in apply_with_input_q_dq_qweight_dq_output_fp32()
78 virtual at::Tensor apply_with_input_q_dq_qweight_dq_relu_output_fp32( in apply_with_input_q_dq_qweight_dq_relu_output_fp32()
95 virtual at::Tensor& apply_dynamic_out( in apply_dynamic_out()
104 virtual at::Tensor& apply_dynamic_relu_out( in apply_dynamic_relu_out()
118 virtual void set_bias(std::optional<at::Tensor> /*bias*/) { in set_bias()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/
H A Dpacked_params.h39 struct LinearPackedParamsBase : public torch::jit::CustomClassHolder { struct
41 LinearPackedParamsBase( in LinearPackedParamsBase() argument
65 virtual void set_bias(const std::optional<at::Tensor>& bias) { in set_bias()