getTensorDescriptor (reference) in projects: aosp_15_r20

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

Searched refs:getTensorDescriptor (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/
H A D	Linear.cpp	178 ….setaMatDesc(cudnn_utils::getTensorDescriptor(input.sizes(), input.strides(), CUDNN_DATA_INT8, 'x'… in apply_impl_helper() 179 ….setbMatDesc(cudnn_utils::getTensorDescriptor(weight_transposed.sizes(), weight_transposed.strides… in apply_impl_helper() 181 ….setcMatDesc(cudnn_utils::getTensorDescriptor(quantized_output.sizes(), quantized_output.strides()… in apply_impl_helper() 198 ….setxDesc(cudnn_utils::getTensorDescriptor(broadcasted_bias.value(), 'b', cudnn_utils::getAlignmen… in apply_impl_helper() 199 ….setbDesc(cudnn_utils::getTensorDescriptor(bias_multiplier_tensor.value(), 'c', cudnn_utils::getAl… in apply_impl_helper() 202 ….setyDesc(cudnn_utils::getTensorDescriptor(broadcasted_bias.value(), 'd', cudnn_utils::getAlignmen… in apply_impl_helper() 216 ….setbDesc(cudnn_utils::getTensorDescriptor(broadcasted_bias.value(), 'n', cudnn_utils::getAlignmen… in apply_impl_helper() 217 ….setyDesc(cudnn_utils::getTensorDescriptor(quantized_output.sizes(), quantized_output.strides(), C… in apply_impl_helper() 232 ….setyDesc(cudnn_utils::getTensorDescriptor(quantized_output.sizes(), quantized_output.strides(), C… in apply_impl_helper() 242 ….setbDesc(cudnn_utils::getTensorDescriptor(requantize_multiplier_tensor, 's', cudnn_utils::getAlig… in apply_impl_helper() [all …]
H A D	Conv.cpp	149 ….setxDesc(cudnn_utils::getTensorDescriptor(input.sizes(), input.strides(), CUDNN_DATA_INT8, 'x', k… in apply_impl_helper() 151 ….setyDesc(cudnn_utils::getTensorDescriptor(quantized_output.sizes(), quantized_output.strides(), C… in apply_impl_helper() 152 ….setwDesc(cudnn_utils::getTensorDescriptor(maybe_padded_weight_.sizes(), maybe_padded_weight_.stri… in apply_impl_helper() 169 ….setxDesc(cudnn_utils::getTensorDescriptor(broadcasted_bias.value(), 'b', cudnn_utils::getAlignmen… in apply_impl_helper() 170 ….setbDesc(cudnn_utils::getTensorDescriptor(bias_multiplier_tensor.value(), 'c', cudnn_utils::getAl… in apply_impl_helper() 171 ….setyDesc(cudnn_utils::getTensorDescriptor(broadcasted_bias.value(), 'd', cudnn_utils::getAlignmen… in apply_impl_helper() 181 ….setbDesc(cudnn_utils::getTensorDescriptor(broadcasted_bias.value(), 'd', cudnn_utils::getAlignmen… in apply_impl_helper() 183 ….setyDesc(cudnn_utils::getTensorDescriptor(quantized_output.sizes(), quantized_output.strides(), C… in apply_impl_helper() 198 ….setyDesc(cudnn_utils::getTensorDescriptor(quantized_output.sizes(), quantized_output.strides(), C… in apply_impl_helper() 208 ….setbDesc(cudnn_utils::getTensorDescriptor(requantize_multiplier_tensor, 's', cudnn_utils::getAlig… in apply_impl_helper() [all …]
H A D	BinaryOps.cpp	171 ….setxDesc(cudnn_utils::getTensorDescriptor(qb.sizes(), qb.strides(), CUDNN_DATA_INT8, 'b', key.inp… in add() 172 ….setbDesc(cudnn_utils::getTensorDescriptor(rhs_multiplier_tensor, 'm', cudnn_utils::getAlignment(r… in add() 173 .setyDesc(cudnn_utils::getTensorDescriptor(add_output, 'c', key.output_alignment)) in add() 181 ….setbDesc(cudnn_utils::getTensorDescriptor(qa.sizes(), qa.strides(), CUDNN_DATA_INT8, 'a', key.inp… in add() 182 .setyDesc(cudnn_utils::getTensorDescriptor(add_output, 'p', key.output_alignment)) in add() 194 .setyDesc(cudnn_utils::getTensorDescriptor(add_output, 'f', key.output_alignment)) in add() 203 ….setbDesc(cudnn_utils::getTensorDescriptor(requantize_multiplier_tensor, 'r', cudnn_utils::getAlig… in add() 204 ….setyDesc(cudnn_utils::getTensorDescriptor(quantized_output.sizes(), quantized_output.strides(), C… in add()
H A D	utils.h	`227 inline cudnn_frontend::Tensor getTensorDescriptor(const at::Tensor &t, int64_t id, uint8_t alignmen… 249 inline cudnn_frontend::Tensor getTensorDescriptor(const c10::IntArrayRef& shape, const c10::IntArra…`
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A D	Conv_v8.cpp	108 cudnn_frontend::Tensor getTensorDescriptor( in getTensorDescriptor() function 440 .setxDesc(getTensorDescriptor( in build_opgraph() 442 .setyDesc(getTensorDescriptor( in build_opgraph() 444 .setwDesc(getTensorDescriptor( in build_opgraph() 495 .setxDesc(getTensorDescriptor( in build_opgraph_fused() 505 .setwDesc(getTensorDescriptor( in build_opgraph_fused() 514 .setbDesc(getTensorDescriptor( in build_opgraph_fused() 532 .setbDesc(getTensorDescriptor( in build_opgraph_fused() 549 .setyDesc(getTensorDescriptor( in build_opgraph_fused()