Home
last modified time | relevance | path

Searched defs:wt_ptr (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DUpSample.cuh324 scalar_t* wt_ptr, in _compute_weights()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqlinear_prepack.cpp175 auto wt_ptr = c10::make_intrusive<PackedLinearWeightsQnnp>( in prepack() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUpSampleKernel.cpp903 scalar_t* wt_ptr = output[3].data_ptr<scalar_t>(); in _compute_index_ranges_weights() local
1308 scalar_t * wt_ptr; in compute_indices_weights() local