Searched refs:_upsample_nearest_exact1d (Results 1 – 11 of 11) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | UpSampleNearest1d.cpp | 37 TORCH_META_FUNC(_upsample_nearest_exact1d) ( in TORCH_META_FUNC() argument
134 Tensor _upsample_nearest_exact1d( in _upsample_nearest_exact1d() function
140 return at::_upsample_nearest_exact1d(input, osize, scale_w); in _upsample_nearest_exact1d()
|
| /external/pytorch/torch/onnx/_internal/exporter/ |
| D | _decomp.py | 50 aten._upsample_nearest_exact1d.vec,
|
| /external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| D | upsampling.h | 156 return torch::_upsample_nearest_exact1d( in interpolate()
|
| /external/pytorch/test/export/ |
| D | testing.py | 82 aten._upsample_nearest_exact1d.vec,
|
| /external/pytorch/aten/src/ATen/native/mps/operations/ |
| D | UpSample.mm | 11 #include <ATen/ops/_upsample_nearest_exact1d.h>
|
| /external/pytorch/test/expect/ |
| D | HasDecompTest.test_aten_core_operators.expect | 25 aten::_upsample_nearest_exact1d.out
|
| /external/pytorch/torch/_decomp/ |
| D | decompositions.py | 2785 @register_decomposition(aten._upsample_nearest_exact1d.vec)
2788 @aten._upsample_nearest_exact1d.vec.py_impl(DispatchKey.CompositeImplicitAutograd)
2789 @aten._upsample_nearest_exact1d.vec.py_impl(DispatchKey.Autograd)
2851 [aten._upsample_nearest_exact1d.default, aten._upsample_nearest_exact1d.out]
2853 @aten._upsample_nearest_exact1d.default.py_impl(DispatchKey.CompositeImplicitAutograd)
2854 @aten._upsample_nearest_exact1d.default.py_impl(DispatchKey.Autograd)
|
| /external/pytorch/aten/src/ATen/ |
| D | autocast_mode.h | 858 _(_upsample_nearest_exact1d) \
|
| /external/pytorch/torch/nn/ |
| D | functional.py | 4541 return torch._C._nn._upsample_nearest_exact1d(input, output_size, scale_factors)
|
| /external/pytorch/torch/_inductor/ |
| D | lowering.py | 3582 @register_lowering(aten._upsample_nearest_exact1d.default)
3583 def _upsample_nearest_exact1d(x, output_size, scales: Optional[float] = None): function
|
| /external/pytorch/torch/ |
| D | _meta_registrations.py | 5743 [aten.upsample_nearest1d.default, aten._upsample_nearest_exact1d.default]
|