Searched refs:bias_multiplier_tensor (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| D | Linear.cpp | 101 std::optional<at::Tensor> bias_multiplier_tensor; in apply_impl_helper() local
111 …bias_multiplier_tensor = at::empty(quantized_output.sizes(), at::device(at::kCUDA).dtype(at::kFloa… in apply_impl_helper()
113 bias_multiplier_tensor.value().fill_(bias_multiplier); in apply_impl_helper()
154 …rs.insert(data_ptrs.end(), {broadcasted_bias.value().data_ptr(), bias_multiplier_tensor.value().da… in apply_impl_helper()
199 …(cudnn_utils::getTensorDescriptor(bias_multiplier_tensor.value(), 'c', cudnn_utils::getAlignment(b… in apply_impl_helper()
203 …dnn_utils::getPointWiseMulDescriptor(at::native::getCudnnDataType(bias_multiplier_tensor.value()))) in apply_impl_helper()
|
| D | Conv.cpp | 73 std::optional<at::Tensor> bias_multiplier_tensor; in apply_impl_helper() local
85 …bias_multiplier_tensor = at::empty(quantized_output.sizes(), at::device(at::kCUDA).dtype(at::kFloa… in apply_impl_helper()
87 bias_multiplier_tensor.value().fill_(bias_multiplier); in apply_impl_helper()
126 …rs.insert(data_ptrs.end(), {broadcasted_bias.value().data_ptr(), bias_multiplier_tensor.value().da… in apply_impl_helper()
170 …(cudnn_utils::getTensorDescriptor(bias_multiplier_tensor.value(), 'c', cudnn_utils::getAlignment(b… in apply_impl_helper()
172 …dnn_utils::getPointWiseMulDescriptor(at::native::getCudnnDataType(bias_multiplier_tensor.value()))) in apply_impl_helper()
|