Searched refs:_upsample_nearest_exact2d (Results 1 – 12 of 12) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | UpSampleNearest2d.cpp | 38 TORCH_META_FUNC(_upsample_nearest_exact2d) ( in TORCH_META_FUNC() argument
159 Tensor _upsample_nearest_exact2d( in _upsample_nearest_exact2d() function
166 return at::_upsample_nearest_exact2d(input, osize, scale_h, scale_w); in _upsample_nearest_exact2d()
|
| /external/pytorch/torch/onnx/_internal/exporter/ |
| D | _decomp.py | 51 aten._upsample_nearest_exact2d.vec,
|
| /external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| D | upsampling.h | 161 return torch::_upsample_nearest_exact2d( in interpolate()
|
| /external/pytorch/test/export/ |
| D | testing.py | 83 aten._upsample_nearest_exact2d.vec,
|
| /external/pytorch/aten/src/ATen/native/mps/operations/ |
| D | UpSample.mm | 15 #include <ATen/ops/_upsample_nearest_exact2d.h>
|
| /external/pytorch/torch/_decomp/ |
| D | decompositions.py | 2786 @register_decomposition(aten._upsample_nearest_exact2d.vec)
2790 @aten._upsample_nearest_exact2d.vec.py_impl(DispatchKey.CompositeImplicitAutograd)
2791 @aten._upsample_nearest_exact2d.vec.py_impl(DispatchKey.Autograd)
2878 [aten._upsample_nearest_exact2d.default, aten._upsample_nearest_exact2d.out]
2880 @aten._upsample_nearest_exact2d.default.py_impl(DispatchKey.CompositeImplicitAutograd)
2881 @aten._upsample_nearest_exact2d.default.py_impl(DispatchKey.Autograd)
2883 def _upsample_nearest_exact2d( function
|
| /external/pytorch/test/expect/ |
| D | HasDecompTest.test_aten_core_operators.expect | 26 aten::_upsample_nearest_exact2d.out
|
| /external/pytorch/aten/src/ATen/ |
| D | autocast_mode.h | 860 _(_upsample_nearest_exact2d) \
|
| /external/pytorch/torch/_inductor/ |
| D | lowering.py | 166 aten._upsample_nearest_exact2d,
3594 @register_lowering(aten._upsample_nearest_exact2d.default)
3595 def _upsample_nearest_exact2d( function
|
| /external/pytorch/torch/nn/ |
| D | functional.py | 4543 return torch._C._nn._upsample_nearest_exact2d(input, output_size, scale_factors)
|
| /external/pytorch/test/cpp/api/ |
| D | functional.cpp | 2333 at::native::_upsample_nearest_exact2d(tensor, osize, torch::nullopt); in TEST_F()
|
| /external/pytorch/torch/ |
| D | _meta_registrations.py | 5759 [aten.upsample_nearest2d.default, aten._upsample_nearest_exact2d.default]
|