Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DResize.cpp22 bool _resize_output_check(const Tensor& output, ArrayRef<T> shape) { in _resize_output_check() function
41 return _resize_output_check(output, shape); in resize_output_check()
45 return _resize_output_check(output, shape); in resize_output_check_symint()
58 if (_resize_output_check<T>(output, shape)) { in _resize_output()
/aosp_15_r20/external/pytorch/torch/_prims_common/
H A Dwrappers.py160 def _resize_output_check(out: TensorLikeType, shape: ShapeType): function
182 if _resize_output_check(out, shape):
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py31 _resize_output_check,
1478 if _resize_output_check(out, B_.shape):