Searched refs:requantize_multiplier_tensor (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| D | BinaryOps.cpp | 118 …at::Tensor requantize_multiplier_tensor = cudnn_utils::getRequantMultiplierTensor(requantize_multi… in add() local
145 … qa.data_ptr<int8_t>(), add_output.data_ptr(), requantize_multiplier_tensor.data_ptr(), in add()
203 …udnn_utils::getTensorDescriptor(requantize_multiplier_tensor, 'r', cudnn_utils::getAlignment(requa… in add()
205 …cudnn_utils::getPointWiseMulDescriptor(at::native::getCudnnDataType(requantize_multiplier_tensor))) in add()
|
| D | Linear.cpp | 100 …at::Tensor requantize_multiplier_tensor = cudnn_utils::getRequantMultiplierTensor(requantize_multi… in apply_impl_helper() local
151 requantize_multiplier_tensor.data_ptr(), quantized_output.data_ptr<int8_t>()}; in apply_impl_helper()
242 …udnn_utils::getTensorDescriptor(requantize_multiplier_tensor, 's', cudnn_utils::getAlignment(requa… in apply_impl_helper()
244 …cudnn_utils::getPointWiseMulDescriptor(at::native::getCudnnDataType(requantize_multiplier_tensor))) in apply_impl_helper()
|
| D | Conv.cpp | 71 …at::Tensor requantize_multiplier_tensor = cudnn_utils::getRequantMultiplierTensor(requantize_multi… in apply_impl_helper() local
123 requantize_multiplier_tensor.data_ptr(), quantized_output.data_ptr<int8_t>()}; in apply_impl_helper()
208 …udnn_utils::getTensorDescriptor(requantize_multiplier_tensor, 's', cudnn_utils::getAlignment(requa… in apply_impl_helper()
210 …cudnn_utils::getPointWiseMulDescriptor(at::native::getCudnnDataType(requantize_multiplier_tensor))) in apply_impl_helper()
|
| D | utils.h | 205 …at::Tensor requantize_multiplier_tensor = at::empty(requantize_multiplier_tensor_size, at::device(… in getRequantMultiplierTensor() local
206 requantize_multiplier_tensor.fill_(requant_multiplier); in getRequantMultiplierTensor()
207 return requantize_multiplier_tensor; in getRequantMultiplierTensor()
|