Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/
Dfbgemm_utils.cpp15 torch::selective_class_<LinearPackedParamsBase>( in register_linear_params()
18 [](const c10::intrusive_ptr<LinearPackedParamsBase>& params) in register_linear_params()
24 LinearPackedParamsBase> { // __setstate__ in register_linear_params()
Dfbgemm_utils.h15 : public LinearPackedParamsBase {
24 : LinearPackedParamsBase( in PackedLinearWeight()
68 static c10::intrusive_ptr<LinearPackedParamsBase> deserialize(
75 static c10::intrusive_ptr<LinearPackedParamsBase> prepack(
Dqnnpack_utils.h18 : public LinearPackedParamsBase {
67 static c10::intrusive_ptr<LinearPackedParamsBase> deserialize(
74 static c10::intrusive_ptr<LinearPackedParamsBase> prepack(
Dqlinear_prepack.cpp53 c10::intrusive_ptr<LinearPackedParamsBase> PackedLinearWeight::
139 c10::intrusive_ptr<LinearPackedParamsBase> PackedLinearWeightQnnp::
156 : LinearPackedParamsBase(out_features_block_size, in_features_block_size), in PackedLinearWeightQnnp()
216 static c10::intrusive_ptr<LinearPackedParamsBase> run( in run()
Dpacked_params.h39 struct LinearPackedParamsBase : public torch::jit::CustomClassHolder { struct
41 LinearPackedParamsBase( in LinearPackedParamsBase() argument
Dqlinear_deserialize.cpp84 c10::intrusive_ptr<LinearPackedParamsBase> PackedLinearWeight::deserialize( in deserialize()
168 c10::intrusive_ptr<LinearPackedParamsBase> PackedLinearWeightQnnp::deserialize( in deserialize()
198 : LinearPackedParamsBase( in PackedLinearWeightQnnp()
Dqlinear_unpack.cpp130 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight) { in run()
Dqlinear_dynamic.cpp170 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight) { in run()
Dqlinear.cpp238 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight, in run()
/external/pytorch/aten/src/ATen/native/quantized/cudnn/
DLinearPrepack.cpp17 c10::intrusive_ptr<LinearPackedParamsBase> PackedLinearWeightCudnn::prepack( in prepack()
43 static c10::intrusive_ptr<LinearPackedParamsBase> run( in run()
Dutils.h27 struct PackedLinearWeightCudnn : public LinearPackedParamsBase {
62 static c10::intrusive_ptr<LinearPackedParamsBase> prepack(
DLinear.cpp347 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight, in run()
/external/pytorch/aten/src/ATen/native/quantized/
Dqlinear_unpack.cpp25 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight) { in run()
33 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight) { in run()
DPackedParams.h6 struct LinearPackedParamsBase : public torch::jit::CustomClassHolder { struct
/external/pytorch/aten/src/ATen/native/quantized/cpu/
Dqlinear_prepack.cpp63 c10::intrusive_ptr<LinearPackedParamsBase> PackedLinearWeight::prepack( in prepack()
138 c10::intrusive_ptr<LinearPackedParamsBase> PackedLinearWeightsQnnp::prepack( in prepack()
188 c10::intrusive_ptr<LinearPackedParamsBase> PackedLinearWeightFp16::prepack( in prepack()
217 c10::intrusive_ptr<LinearPackedParamsBase> PackedLinearWeightsOnednn::prepack( in prepack()
384 .typed<c10::intrusive_ptr<LinearPackedParamsBase>( in wrapped_quantized_linear()
462 .typed<c10::intrusive_ptr<LinearPackedParamsBase>( in _wrapped_linear_prepack()
498 cpp_custom_type_hack::cast<c10::intrusive_ptr<LinearPackedParamsBase>>( in _wrapped_quantized_linear_prepacked()
566 static c10::intrusive_ptr<LinearPackedParamsBase> run( in run()
597 static c10::intrusive_ptr<LinearPackedParamsBase> run( in run()
Dfbgemm_utils.cpp438 torch::selective_class_<LinearPackedParamsBase>( in register_linear_params()
441 [](const c10::intrusive_ptr<LinearPackedParamsBase>& params) in register_linear_params()
447 LinearPackedParamsBase> { // __setstate__ in register_linear_params()
494 .def("bias", [](const c10::intrusive_ptr<LinearPackedParamsBase>& self) { in register_linear_params()
497 .def("unpack", &LinearPackedParamsBase::unpack); in register_linear_params()
Dfbgemm_utils.h25 struct TORCH_API PackedLinearWeight : public LinearPackedParamsBase {
90 static c10::intrusive_ptr<LinearPackedParamsBase> prepack(
112 struct TORCH_API PackedLinearWeightFp16 : public LinearPackedParamsBase {
154 static c10::intrusive_ptr<LinearPackedParamsBase> prepack(
Dqlinear_dynamic.cpp630 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight, in run()
646 const c10::intrusive_ptr<LinearPackedParamsBase>& packed_weight) { in run()
665 const c10::intrusive_ptr<LinearPackedParamsBase>& /* packed_weight */) {
DOnednnUtils.h119 struct PackedLinearWeightsOnednn : public LinearPackedParamsBase {
165 static c10::intrusive_ptr<LinearPackedParamsBase> prepack(
DQnnpackUtils.h36 struct PackedLinearWeightsQnnp : public LinearPackedParamsBase {
85 static c10::intrusive_ptr<LinearPackedParamsBase> prepack(
Dqlinear.cpp1113 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()
/external/pytorch/aten/src/ATen/native/
DRNN.cpp148 std::vector<c10::intrusive_ptr<LinearPackedParamsBase>>>;
381 c10::intrusive_ptr<LinearPackedParamsBase> w_ih_packed,
382 c10::intrusive_ptr<LinearPackedParamsBase> w_hh_packed,
389 c10::intrusive_ptr<LinearPackedParamsBase> in QuantizedCellParamsDynamic()
391 c10::intrusive_ptr<LinearPackedParamsBase> in QuantizedCellParamsDynamic()
402 c10::intrusive_ptr<LinearPackedParamsBase> packed_w_ih;
403 c10::intrusive_ptr<LinearPackedParamsBase> packed_w_hh;
434 std::vector<c10::intrusive_ptr<LinearPackedParamsBase>> in __getstate__()
463 c10::intrusive_ptr<LinearPackedParamsBase> w_ih_packed, in make_quantized_cell_params_dynamic()
464 c10::intrusive_ptr<LinearPackedParamsBase> w_hh_packed, in make_quantized_cell_params_dynamic()
[all …]
DQuantizedLinear.cpp35 CAFFE_KNOWN_TYPE(c10::intrusive_ptr<LinearPackedParamsBase>);
406 c10::intrusive_ptr<LinearPackedParamsBase> packed_weight = in fbgemm_pack_gemm_matrix_fp16()
433 c10::intrusive_ptr<LinearPackedParamsBase>>(packed_weight)) in fbgemm_linear_fp16_weight_fp32_activation()
/external/pytorch/torch/csrc/jit/runtime/static/
Dops.cpp2221 c10::intrusive_ptr<LinearPackedParamsBase> packed_weight; in __anon5bf8650b6e02()
2223 packed_weight = w->toCustomClass<LinearPackedParamsBase>(); in __anon5bf8650b6e02()
2249 p_node->Input(1).toCustomClass<LinearPackedParamsBase>(); in __anon5bf8650b6e02()
2266 c10::intrusive_ptr<LinearPackedParamsBase> packed_weight; in __anon5bf8650b7002()
2268 packed_weight = w->toCustomClass<LinearPackedParamsBase>(); in __anon5bf8650b7002()
2296 p_node->Input(1).toCustomClass<LinearPackedParamsBase>(); in __anon5bf8650b7002()
2307 c10::intrusive_ptr<LinearPackedParamsBase> packed_weight,
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()
[all …]
/external/pytorch/torch/csrc/jit/tensorexpr/
Dexternal_functions.cpp619 reinterpret_cast<LinearPackedParamsBase*>(buf_data[2]); in nnc_aten_quantized_linear()
650 reinterpret_cast<LinearPackedParamsBase*>(buf_data[2]); in nnc_aten_quantized_linear_out()
681 reinterpret_cast<LinearPackedParamsBase*>(buf_data[2]); in nnc_aten_quantized_linear_relu()