Searched refs:nearest_neighbor_exact_compute_source_index (Results 1 – 7 of 7) sorted by relevance
322 inline int64_t nearest_neighbor_exact_compute_source_index( in nearest_neighbor_exact_compute_source_index() function361 return nearest_neighbor_exact_compute_source_index(scale, output_index, input_size); in nearest_exact_idx()
218 …return _upsample_nearest2d_quantized_cpu<nearest_neighbor_exact_compute_source_index>(input, osize… in _upsample_nearest_exact2d_quantized_cpu()
233 return _upsample_nearest3d_quantized_cpu<nearest_neighbor_exact_compute_source_index>( in _upsample_nearest_exact3d_quantized_cpu()
216 …upsample_nearest1d_out_cuda_template<nearest_neighbor_exact_compute_source_index>(output, input, o… in TORCH_IMPL_FUNC()
147 __device__ __forceinline__ int nearest_neighbor_exact_compute_source_index( in nearest_neighbor_exact_compute_source_index() function
313 …upsample_nearest3d_out_cuda_template<nearest_neighbor_exact_compute_source_index>(output, input, o… in TORCH_IMPL_FUNC()
462 upsample_nearest2d_out_cuda_template<nearest_neighbor_exact_compute_source_index>( in TORCH_IMPL_FUNC()