| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | UpSampleNearest3d.cpp | 15 #include <ATen/ops/upsample_nearest3d.h>
23 TORCH_META_FUNC(upsample_nearest3d) ( in TORCH_META_FUNC() argument
164 Tensor upsample_nearest3d( in upsample_nearest3d() function
172 return at::upsample_nearest3d(input, osize, scale_d, scale_h, scale_w); in upsample_nearest3d()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | UpSampleNearest3d.cu | 17 #include <ATen/ops/upsample_nearest3d.h>
178 // upsample_nearest3d meta call makes sure `nbatch != 0` in upsample_nearest3d_out_cuda_template()
185 …"upsample_nearest3d only supports output tensors with less than INT_MAX elements, but got ", outpu… in upsample_nearest3d_out_cuda_template()
252 // upsample_nearest3d meta call makes sure `nbatch != 0` in upsample_nearest3d_backward_out_cuda_template()
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/ |
| H A D | _decomp.py | 63 aten.upsample_nearest3d.default,
64 aten.upsample_nearest3d.vec,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | UpSampleNearest3d.cpp | 167 AT_DISPATCH_QINT_TYPES(input.scalar_type(), "upsample_nearest3d", [&] { in _upsample_nearest3d_quantized_cpu()
195 AT_DISPATCH_QINT_TYPES(input_contig.scalar_type(), "upsample_nearest3d", [&] { in _upsample_nearest3d_quantized_cpu()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesDecompositions.cpp | 346 m.impl("upsample_nearest3d.vec", native::upsample_nearest3d); in TORCH_LIBRARY_IMPL()
|
| H A D | BatchRulesModules.cpp | 389 EXISTING_BDIM(upsample_nearest3d); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset8.py | 78 "aten::upsample_nearest3d",
79 decorate=[symbolic_helper._apply_params("upsample_nearest3d", 5, "nearest")],
|
| H A D | symbolic_opset10.py | 439 "aten::upsample_nearest3d",
440 decorate=[symbolic_helper._apply_params("upsample_nearest3d", 5, "nearest")],
|
| H A D | symbolic_opset11.py | 343 "aten::upsample_nearest3d",
344 decorate=[symbolic_helper._apply_params("upsample_nearest3d", 5, "nearest")],
|
| H A D | symbolic_opset9.py | 1864 "aten::upsample_nearest3d",
1866 symbolic_helper._apply_params("upsample_nearest3d", 5, "nearest"),
1867 _export("upsample_nearest3d"),
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | quantization_patterns.h | 848 "aten::upsample_nearest3d", {"%output_size", "%scale_factors"}); in quant_fusion_pattern_and_replacements()
868 auto upsample_nearest3d = getInputTensorQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
869 "aten::upsample_nearest3d", in quant_fusion_pattern_and_replacements()
1058 std::move(upsample_nearest3d), in quant_fusion_pattern_and_replacements()
|
| H A D | helper.cpp | 154 "upsample_nearest3d",
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 2766 @register_decomposition(aten.upsample_nearest3d.vec)
2771 @aten.upsample_nearest3d.vec.py_impl(DispatchKey.CompositeImplicitAutograd)
2772 @aten.upsample_nearest3d.vec.py_impl(DispatchKey.Autograd)
2892 @register_decomposition([aten.upsample_nearest3d.default, aten.upsample_nearest3d.out])
2893 @aten.upsample_nearest3d.default.py_impl(DispatchKey.CompositeImplicitAutograd)
2894 @aten.upsample_nearest3d.default.py_impl(DispatchKey.Autograd)
2896 def upsample_nearest3d( function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/mps/ |
| H A D | MPSFallback.mm | 94 m.impl("upsample_nearest3d.vec", torch::CppFunction::makeFromBoxedFunction<&mps_fallback>());
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 624 - aten::upsample_nearest3d.vec
1006 aten::upsample_nearest3d.vec: 2
|
| H A D | model_ops.yaml | 384 aten::upsample_nearest3d.vec: 2
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/utils/ |
| H A D | op_registry.cpp | 52 …"aten::upsample_nearest3d(Tensor self, int[] output_size, float? scales_d, float? scales_h, float?… in nn_ops_first_input_preserving()
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | testing.py | 81 aten.upsample_nearest3d.vec,
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | annotated_ops | 483 upsample_nearest3d, misc
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_meta.py | 849 aten.upsample_nearest3d.vec : {bf16, f32, f64, u8},
900 aten.upsample_nearest3d.vec: {f16}, # aten::upsample_nearest3d.vec
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_dtype_analysis.py | 83 "upsample_nearest3d",
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | upsampling.h | 149 return torch::upsample_nearest3d( in interpolate()
|
| /aosp_15_r20/external/pytorch/test/expect/ |
| H A D | HasDecompTest.test_aten_core_operators.expect | 518 aten::upsample_nearest3d.out
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | register_prim_ops_fulljit.cpp | 521 return at::upsample_nearest3d( in interpolate()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | autocast_mode.h | 861 _(upsample_nearest3d) \
|