Searched refs:real_input_index (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | UpSample.h | 437 inline void guard_index_and_lambda(const opmath_t& real_input_index, const int64_t& input_size, int… in guard_index_and_lambda() argument
438 input_index = std::min(static_cast<int64_t>(floorf(real_input_index)), input_size - 1); in guard_index_and_lambda()
440 std::max(real_input_index - input_index, static_cast<opmath_t>(0)), in guard_index_and_lambda()
463 const auto real_input_index = in compute_source_index_and_lambda() local
466 guard_index_and_lambda(real_input_index, input_size, input_index0, lambda1); in compute_source_index_and_lambda()
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | UpSampleKernel.cpp | 802 const auto real_input_index = area_pixel_compute_source_index<scalar_t>( in _compute_indices_min_size_weights() local
807 guard_index_and_lambda(real_input_index, input_size, input_index, lambda); in _compute_indices_min_size_weights()
1092 const auto real_input_index = in compute_indices_weights() local
1095 input_index = static_cast<int64_t>(floorf(real_input_index)); in compute_indices_weights()
1142 const auto real_input_index = in compute_indices_weights() local
1145 input_index = static_cast<int64_t>(floorf(real_input_index + 0.5)); in compute_indices_weights()
1310 const auto real_input_index = in compute_indices_weights() local
1314 guard_index_and_lambda(real_input_index, input_size, input_index, lambda); in compute_indices_weights()
|