| /aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/ |
| H A D | decomposition_skip.py | 126 op_callable = torch._C._nn.upsample_trilinear3d # type: ignore[attr-defined]
128 new_op_name = "upsample_trilinear3d"
137 …torch._C._nn.upsample_trilinear3d = torch.ops.onnx_export.upsample_trilinear3d # type: ignore[att…
149 torch._C._nn.upsample_trilinear3d = cls.op_callable # type: ignore[attr-defined]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | UpSampleTrilinear3d.cpp | 14 #include <ATen/ops/upsample_trilinear3d.h>
22 TORCH_META_FUNC(upsample_trilinear3d) ( in TORCH_META_FUNC() argument
101 Tensor upsample_trilinear3d( in upsample_trilinear3d() function
110 return at::upsample_trilinear3d(input, osize, align_corners, scale_d, scale_h, scale_w); in upsample_trilinear3d()
|
| H A D | native_functions.yaml | 12500 - func: upsample_trilinear3d.vec(Tensor input, SymInt[]? output_size, bool align_corners, float[]? …
12502 autogen: upsample_trilinear3d.vec_out
12654 - func: upsample_trilinear3d.out(Tensor self, SymInt[3] output_size, bool align_corners, float? sca…
12661 - func: upsample_trilinear3d(Tensor self, SymInt[3] output_size, bool align_corners, float? scales_…
12663 structured_delegate: upsample_trilinear3d.out
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/ |
| H A D | _decomp.py | 65 aten.upsample_trilinear3d.default,
66 aten.upsample_trilinear3d.vec,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesDecompositions.cpp | 347 m.impl("upsample_trilinear3d.vec", native::upsample_trilinear3d); in TORCH_LIBRARY_IMPL()
|
| H A D | BatchRulesModules.cpp | 390 EXISTING_BDIM(upsample_trilinear3d); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset8.py | 90 "aten::upsample_trilinear3d",
91 decorate=[symbolic_helper._apply_params("upsample_trilinear3d", 5, "linear")],
|
| H A D | symbolic_opset10.py | 451 "aten::upsample_trilinear3d",
452 decorate=[symbolic_helper._apply_params("upsample_trilinear3d", 5, "linear")],
|
| H A D | symbolic_opset11.py | 355 "aten::upsample_trilinear3d",
356 decorate=[symbolic_helper._apply_params("upsample_trilinear3d", 5, "linear")],
|
| H A D | symbolic_opset9.py | 1885 "aten::upsample_trilinear3d",
1887 symbolic_helper._apply_params("upsample_trilinear3d", 5, "linear"),
1888 _export("upsample_trilinear3d"),
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | quantization_patterns.h | 859 "aten::upsample_trilinear3d", in quant_fusion_pattern_and_replacements()
879 auto upsample_trilinear3d = getInputTensorQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
880 "aten::upsample_trilinear3d", in quant_fusion_pattern_and_replacements()
1061 std::move(upsample_trilinear3d), in quant_fusion_pattern_and_replacements()
|
| H A D | helper.cpp | 157 "upsample_trilinear3d",
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 625 - aten::upsample_trilinear3d.vec
1007 aten::upsample_trilinear3d.vec: 1
|
| H A D | model_ops.yaml | 385 aten::upsample_trilinear3d.vec: 1
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/utils/ |
| H A D | op_registry.cpp | 53 …"aten::upsample_trilinear3d(Tensor self, int[] output_size, bool align_corners, float? scales_d, f… in nn_ops_first_input_preserving()
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | testing.py | 96 aten.upsample_trilinear3d.vec,
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 3628 @register_decomposition(aten.upsample_trilinear3d.vec)
3633 @aten.upsample_trilinear3d.vec.py_impl(DispatchKey.CompositeImplicitAutograd)
3634 @aten.upsample_trilinear3d.vec.py_impl(DispatchKey.Autograd)
3668 [aten.upsample_trilinear3d.default, aten.upsample_trilinear3d.out]
3671 def upsample_trilinear3d( function
|
| H A D | __init__.py | 472 aten.upsample_trilinear3d,
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | annotated_ops | 479 upsample_trilinear3d, misc
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_dtype_analysis.py | 84 "upsample_trilinear3d",
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | upsampling.h | 222 return torch::upsample_trilinear3d( in interpolate()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | register_prim_ops_fulljit.cpp | 573 return at::upsample_trilinear3d( in interpolate()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | autocast_mode.h | 866 _(upsample_trilinear3d) \
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | conf.py | 1723 "upsample_trilinear3d",
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | UpSampleKernel.cpp | 126 // There is an unexpected 2x slowdown for upsample_trilinear3d channels_first
|