Searched refs:apply_dynamic_impl (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | qlinear_dynamic.cpp | 35 at::Tensor PackedLinearWeight::apply_dynamic_impl( in apply_dynamic_impl() function in PackedLinearWeight
233 return apply_dynamic_impl</*ReluFused=*/false>( in apply_dynamic()
240 return apply_dynamic_impl</*ReluFused=*/true>(std::move(input), reduce_range); in apply_dynamic_relu()
247 at::Tensor PackedLinearWeightsQnnp::apply_dynamic_impl( in apply_dynamic_impl() function in PackedLinearWeightsQnnp
396 return apply_dynamic_impl</*ReluFused=*/false>(std::move(input), reduce_range); in apply_dynamic()
402 return apply_dynamic_impl</*ReluFused=*/true>(std::move(input), reduce_range); in apply_dynamic_relu()
410 at::Tensor& PackedLinearWeightFp16::apply_dynamic_impl( in apply_dynamic_impl() function in PackedLinearWeightFp16
461 return apply_dynamic_impl</*ReluFused=*/false>(input, output); in apply_dynamic()
468 return apply_dynamic_impl</*ReluFused=*/true>(input, output); in apply_dynamic_relu()
476 return apply_dynamic_impl<false>(input, output); in apply_dynamic_out()
[all …]
|
| D | fbgemm_utils.h | 109 at::Tensor apply_dynamic_impl(at::Tensor input, bool reduce_range = false);
162 at::Tensor& apply_dynamic_impl(const at::Tensor& input, at::Tensor& output);
|
| D | OnednnUtils.h | 181 at::Tensor apply_dynamic_impl(at::Tensor input, bool reduce_range=false);
|
| D | QnnpackUtils.h | 113 at::Tensor apply_dynamic_impl(at::Tensor input, bool reduce_range);
|
| /external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/ |
| D | qlinear_dynamic.cpp | 28 at::Tensor PackedLinearWeightQnnp::apply_dynamic_impl<true>( in apply_dynamic_impl() function in ao::sparse::PackedLinearWeightQnnp
38 at::Tensor PackedLinearWeightQnnp::apply_dynamic_impl<false>( in apply_dynamic_impl() function in ao::sparse::PackedLinearWeightQnnp
153 return apply_dynamic_impl<false>(input); in apply_dynamic()
158 return apply_dynamic_impl<true>(input); in apply_dynamic_relu()
|
| D | qnnpack_utils.h | 87 at::Tensor apply_dynamic_impl(const at::Tensor& input);
|