Home
last modified time | relevance | path

Searched defs:padding_mode (Results 1 – 25 of 40) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/qat/modules/
H A Dconv_fused.py56 padding_mode, argument
470 padding_mode="zeros", argument
541 padding_mode="zeros", argument
603 padding_mode="zeros", argument
667 padding_mode="zeros", argument
738 padding_mode="zeros", argument
800 padding_mode="zeros", argument
864 padding_mode="zeros", argument
934 padding_mode="zeros", argument
998 padding_mode="zeros", argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DGridSampler.cpp20 int64_t interpolation_mode, int64_t padding_mode, in grid_sampler_2d_cuda()
32 int64_t interpolation_mode, int64_t padding_mode, in grid_sampler_3d_cuda()
46 const Tensor& grid, int64_t interpolation_mode, int64_t padding_mode, in grid_sampler_2d_backward_cuda()
65 const Tensor& grid, int64_t interpolation_mode, int64_t padding_mode, in grid_sampler_3d_backward_cuda()
H A DGridSampler.cuh151 GridSamplerPadding padding_mode, in compute_coordinates()
177 GridSamplerPadding padding_mode, in grid_sampler_compute_source_index()
193 GridSamplerPadding padding_mode, in grid_sampler_compute_source_index_set_grad()
232 GridSamplerPadding padding_mode, in get_value_bounded()
284 GridSamplerPadding padding_mode, in add_value_bounded()
H A DGridSampler.cu32 const GridSamplerPadding padding_mode, in grid_sampler_2d_kernel()
162 const GridSamplerPadding padding_mode, in grid_sampler_3d_kernel()
320 const GridSamplerPadding padding_mode, in grid_sampler_2d_backward_kernel()
526 const GridSamplerPadding padding_mode, in grid_sampler_3d_backward_kernel()
752 int64_t interpolation_mode, int64_t padding_mode, bool align_corners) { in launch_grid_sampler_2d_forward_kernel()
796 int64_t interpolation_mode, int64_t padding_mode, bool align_corners) { in launch_grid_sampler_3d_forward_kernel()
842 const TensorBase &grid, int64_t interpolation_mode, int64_t padding_mode, in launch_grid_sampler_2d_backward_kernel()
905 const TensorBase& grid, int64_t interpolation_mode, int64_t padding_mode, in launch_grid_sampler_3d_backward_kernel()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DGridSampler.h143 GridSamplerPadding padding_mode, in compute_coordinates()
166 GridSamplerPadding padding_mode, in grid_sampler_compute_source_index()
181 GridSamplerPadding padding_mode, in grid_sampler_compute_source_index_set_grad()
221 GridSamplerPadding padding_mode, in get_value_bounded()
265 GridSamplerPadding padding_mode, in add_value_bounded()
H A DGridSampler.cpp43 GridSamplerPadding padding_mode, in grid_sampler_3d_cpu_impl()
208 GridSamplerPadding padding_mode, in grid_sampler_3d_backward_cpu_impl()
567 auto padding_mode = static_cast<GridSamplerPadding>(padding_mode_); in _grid_sampler_2d_cpu_fallback() local
723 const auto padding_mode = static_cast<GridSamplerPadding>(padding_mode_); in _grid_sampler_2d_cpu_fallback_backward() local
920 int64_t interpolation_mode, int64_t padding_mode, in grid_sampler_2d_cpu()
964 int64_t interpolation_mode, int64_t padding_mode, in grid_sampler_3d_cpu()
980 int64_t interpolation_mode, int64_t padding_mode, bool align_corners, in grid_sampler_2d_backward_cpu()
1029 int64_t interpolation_mode, int64_t padding_mode, bool align_corners, in grid_sampler_3d_backward_cpu()
1050 int64_t padding_mode, in grid_sampler()
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/
H A Dconv.py51 padding_mode="zeros", argument
70 padding_mode="zeros", argument
537 padding_mode="zeros", argument
666 padding_mode="zeros", argument
761 padding_mode, argument
930 padding_mode="zeros", argument
1053 padding_mode="zeros", argument
1178 padding_mode="zeros", argument
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/
H A Dconv.py149 padding_mode="zeros", argument
233 padding_mode="zeros", argument
324 padding_mode="zeros", argument
406 padding_mode="zeros", argument
488 padding_mode="zeros", argument
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/reference/modules/
H A Dconv.py128 padding_mode="zeros", argument
191 padding_mode="zeros", argument
365 padding_mode="zeros", argument
446 padding_mode="zeros", argument
/aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/quantized/modules/
H A Dconv_relu.py43 padding_mode="zeros", argument
124 padding_mode="zeros", argument
205 padding_mode="zeros", argument
H A Dconv_add.py34 padding_mode="zeros", argument
102 padding_mode="zeros", argument
/aosp_15_r20/external/pytorch/test/quantization/eager/
H A Dtest_quantize_eager_qat.py76 padding_mode, argument
237 padding_mode='zeros', argument
773 padding_mode, argument
925 padding_mode, argument
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/
H A Dfunctional.py181 padding_mode="zeros", argument
253 padding_mode="zeros", argument
325 padding_mode="zeros", argument
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_onnx_opset.py560 def __init__(self, mode, padding_mode, align_corners): argument
583 ops, opset_version, mode, padding_mode, align_corners, x_shape, grid argument
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DPadNd.h7 enum class padding_mode { enum
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dvision.h61 GridSampleFuncOptions::padding_mode_t padding_mode, in grid_sample()
/aosp_15_r20/hardware/interfaces/security/keymint/aidl/vts/functional/
H A DKeyMintTest.cpp686 for (auto padding_mode : ValidPaddingModes(Algorithm::AES, block_mode)) { in TEST_P() local
725 for (auto padding_mode : ValidPaddingModes(Algorithm::AES, block_mode)) { in TEST_P() local
745 for (auto padding_mode : ValidPaddingModes(Algorithm::AES, block_mode)) { in TEST_P() local
773 for (auto padding_mode : InvalidPaddingModes(Algorithm::AES, block_mode)) { in TEST_P() local
817 for (auto padding_mode : ValidPaddingModes(Algorithm::AES, block_mode)) { in TEST_P() local
844 for (auto padding_mode : ValidPaddingModes(Algorithm::AES, block_mode)) { in TEST_P() local
871 for (auto padding_mode : ValidPaddingModes(Algorithm::AES, block_mode)) { in TEST_P() local
911 for (auto padding_mode : ValidPaddingModes(Algorithm::AES, block_mode)) { in TEST_P() local
954 for (auto padding_mode : ValidPaddingModes(Algorithm::AES, block_mode)) { in TEST_P() local
973 for (auto padding_mode : ValidPaddingModes(Algorithm::AES, block_mode)) { in TEST_P() local
[all …]
H A DKeyMintAidlTestBase.cpp917 PaddingMode padding_mode, const string& iv, in AesCheckEncryptOneByteAtATime()
935 PaddingMode padding_mode, const string& iv, in CheckEncryptOneByteAtATime()
1049 PaddingMode padding_mode, const string& key, in CheckTripleDesTestVector()
1416 PaddingMode padding_mode, const vector<uint8_t>& iv) { in DecryptMessage()
/aosp_15_r20/external/ComputeLibrary/tests/validate_examples/
H A Dgraph_convolution.cpp235 EnumOption<ConvolutionPaddingMode> *padding_mode; /**< Padding mode */ member in __anon238537030111::ConvolutionOptions
H A Dgraph_depthwiseconvolution.cpp235 …EnumOption<ConvolutionPaddingMode> *padding_mode; /**< Padding mode … member in __anonf2723d900111::DepthConvolutionOptions
/aosp_15_r20/hardware/interfaces/keymaster/4.0/vts/functional/
H A DKeymasterHidlTest.cpp580 PaddingMode padding_mode, const string& key, in CheckTripleDesTestVector()
714 PaddingMode padding_mode, const HidlBuf& iv) { in DecryptMessage()
/aosp_15_r20/external/rust/android-crates-io/crates/mls-rs/src/group/
Dmls_rules.rs95 pub padding_mode: PaddingMode, field
/aosp_15_r20/system/security/keystore2/test_utils/
H A Dauthorizations.rs124 pub fn padding_mode(mut self, p: PaddingMode) -> Self { in padding_mode() method
/aosp_15_r20/external/executorch/backends/xnnpack/test/ops/
H A Dconv2d.py31 padding_mode="zeros", argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DGridSamplerKernel.cpp1146 int64_t interpolation_mode, int64_t padding_mode, bool align_corners) { in grid_sampler_2d_cpu_kernel_impl()
1216 int64_t padding_mode, in grid_sampler_2d_backward_cpu_kernel_impl()

12