Home
last modified time | relevance | path

Searched defs:padding_arg (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DPool.cpp91 const IntArrayRef padding_arg, in pool2d()
232 const IntArrayRef padding_arg, in avg_pool2d()
250 const IntArrayRef padding_arg, in max_pool2d()
H A DConvolution.cpp922 const IntArrayRef padding_arg, in Conv2dPackedContext()
1257 const IntArrayRef padding_arg, in create()
1331 const IntArrayRef padding_arg, in Conv1dPackedContext()
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Datrous_convolution_test.py217 …def combined_op(converted_input, num_spatial_dims, padding_arg): # pylint: disable=unused-argument argument