Searched refs:check_valid_input (Results 1 – 4 of 4) sorted by relevance
37 at::native::padding::check_valid_input<1>(input, paddingSize); in TORCH_META_FUNC()99 at::native::padding::check_valid_input<2>(input, paddingSize); in TORCH_META_FUNC()142 at::native::padding::check_valid_input<3>(input, paddingSize); in TORCH_META_FUNC()250 at::native::padding::check_valid_input<3>(input, paddingSize); in replication_pad3d_backward_out_cpu_template()
38 at::native::padding::check_valid_input<1>(input, padding); in TORCH_META_FUNC()118 at::native::padding::check_valid_input<3>(input, padding); in TORCH_META_FUNC()219 at::native::padding::check_valid_input<2>(input, padding); in reflection_pad2d_out_template()
29 inline void check_valid_input(const Tensor& input, IntArrayRef padding) { in check_valid_input() function
282 at::native::padding::check_valid_input<2>(input_, padding); in reflection_pad2d_out_template()