Searched refs:nn_compute_source_index_fn_t (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | UpSampleNearest2d.cpp | 26 typedef int64_t (*nn_compute_source_index_fn_t)(const float, int64_t, int64_t); typedef
29 template <typename scalar_t, nn_compute_source_index_fn_t nn_compute_source_index_fn>
85 template <typename scalar_t, nn_compute_source_index_fn_t nn_compute_source_index_fn>
120 template <nn_compute_source_index_fn_t nn_compute_source_index_fn>
|
| H A D | UpSampleNearest3d.cpp | 24 typedef int64_t (*nn_compute_source_index_fn_t)(const float, int64_t, int64_t); typedef
27 template <typename scalar_t, nn_compute_source_index_fn_t nn_compute_source_index_fn>
84 template <typename scalar_t, nn_compute_source_index_fn_t nn_compute_source_index_fn>
132 template <nn_compute_source_index_fn_t nn_compute_source_index_fn>
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | UpSampleNearest1d.cu | 28 typedef int (*nn_compute_source_index_fn_t)(const float, int, int); typedef
36 template <typename scalar_t, nn_compute_source_index_fn_t nn_compute_source_index_fn>
102 template<nn_compute_source_index_fn_t nn_compute_source_index_fn>
146 template<nn_compute_source_index_fn_t nn_bw_compute_source_index_fn>
|
| H A D | UpSampleNearest2d.cu | 32 typedef int (*nn_compute_source_index_fn_t)(const float, int, int); typedef
39 template <typename scalar_t, nn_compute_source_index_fn_t nn_compute_source_index_fn>
82 template <typename scalar_t, nn_compute_source_index_fn_t nn_compute_source_index_fn>
205 template<nn_compute_source_index_fn_t nn_compute_source_index_fn>
|
| H A D | UpSampleNearest3d.cu | 34 typedef int (*nn_compute_source_index_fn_t)(const float, int, int); typedef
41 template <typename scalar_t, nn_compute_source_index_fn_t nn_compute_source_index_fn>
146 template<nn_compute_source_index_fn_t nn_compute_source_index_fn>
|