Home
last modified time | relevance | path

Searched refs:generate_intervals (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/native/
DFractionalMaxPooling.h9 inline std::vector<int64_t> generate_intervals( in generate_intervals() function
DFractionalMaxPool3d.cpp119 auto sequenceT = generate_intervals<scalar_t>( in fractional_max_pool3d_out_single_batch_frame()
121 auto sequenceH = generate_intervals<scalar_t>( in fractional_max_pool3d_out_single_batch_frame()
123 auto sequenceW = generate_intervals<scalar_t>( in fractional_max_pool3d_out_single_batch_frame()
DFractionalMaxPool2d.cpp145 auto sequenceW = generate_intervals<scalar_t>( in fractional_max_pool2d_out_single_batch_frame()
147 auto sequenceH = generate_intervals<scalar_t>( in fractional_max_pool2d_out_single_batch_frame()