Home
last modified time | relevance | path

Searched defs:output_mask (Results 1 – 25 of 49) sorted by relevance

12

/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_convolution_backward.cpp37 ET_UNUSED exec_aten::ArrayRef<bool> output_mask, in check_convolution_backward_args()
102 exec_aten::ArrayRef<bool> output_mask, in conv2d_backward_impl()
241 exec_aten::ArrayRef<bool> output_mask, in convolution_backward_out()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DGridSampler.cpp47 bool align_corners, std::array<bool, 2> output_mask) { in grid_sampler_2d_backward_cuda()
66 bool align_corners, std::array<bool,2> output_mask) { in grid_sampler_3d_backward_cuda()
H A DNaiveDilatedConvolution.cu458 const std::array<bool, 3ul> output_mask) { in slow_conv_dilated2d_backward_cuda()
564 const std::array<bool, 3ul> output_mask) { in slow_conv_dilated3d_backward_cuda()
H A DDepthwiseConv3d.cu538 const std::array<bool, 3> output_mask) in _depthwise_3d_backward_cuda_out()
673 const std::array<bool, 3> output_mask) { in conv_depthwise3d_backward_cuda()
H A DGridSampler.cu843 bool align_corners, std::array<bool,2> output_mask) { in launch_grid_sampler_2d_backward_kernel()
906 bool align_corners, std::array<bool,2> output_mask) { in launch_grid_sampler_3d_backward_kernel()
H A DConvolutionMM2d.cu473 std::array<bool, 3> output_mask) { in slow_conv2d_backward_cuda()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesNorm.cpp209 std::array<bool,3> output_mask) { in batch_norm_backward_plumbing()
376 int64_t N, int64_t C, int64_t HxW, int64_t group, std::array<bool,3> output_mask in native_group_norm_backward_plumbing()
578 std::array<bool,3> output_mask) { in native_layer_norm_backward_plumbing()
827 std::array<bool,3> output_mask) { in cudnn_batch_norm_backward_wrapper()
843 std::array<bool,3> output_mask) { in miopen_batch_norm_backward_wrapper()
H A DBatchRulesConvolution.cpp142 const Tensor& grad_output_, std::array<bool, 3> output_mask) { in compute_grad_bias()
365 c10::SymIntArrayRef output_padding, c10::SymInt groups, std::array<bool, 3> output_mask) { in convolution_backward_plumbing()
/aosp_15_r20/external/pytorch/test/
H A Dtest_meta.py1383 def _norm_backwards_test_helper(self, op, args, output_mask, expected_shapes): argument
1415 def test_layer_norm_backward(self, output_mask): argument
1448 def test_group_norm_backward(self, output_mask): argument
1479 def test_batch_norm_backward(self, output_mask): argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNaiveDilatedConvolution.cpp638 const std::array<bool, 3ul> output_mask) { in slow_conv_dilated2d_backward_cpu()
697 const std::array<bool, 3ul> output_mask) { in slow_conv_dilated3d_backward_cpu()
H A DConvolution.cpp1724 … bool transposed, IntArrayRef output_padding, int64_t groups, std::array<bool, 3> output_mask) { in convolution_backward_overrideable()
1742 std::array<bool, 3> output_mask) { in _convolution_double_backward()
1940 const std::array<bool, 3> output_mask, in _convolution_backward_nogroup_backend()
1996 int64_t groups, std::array<bool, 3> output_mask) { in convolution_backward()
H A DGridSampler.cpp209 bool align_corners, std::array<bool,2> output_mask) { in grid_sampler_3d_backward_cpu_impl()
981 std::array<bool,2> output_mask) { in grid_sampler_2d_backward_cpu()
1030 std::array<bool,2> output_mask) { in grid_sampler_3d_backward_cpu()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DConv_miopen.cpp62 bool benchmark, bool deterministic, std::array<bool,3> output_mask) { in miopen_convolution_backward()
90 bool benchmark, bool deterministic, std::array<bool,3> output_mask) { in miopen_convolution_transpose_backward()
118 bool benchmark, bool deterministic, std::array<bool,3> output_mask) { in miopen_depthwise_convolution_backward()
1197 bool benchmark, bool deterministic, std::array<bool,3> output_mask) { in miopen_convolution_transpose_backward()
1426 bool benchmark, bool deterministic, std::array<bool,3> output_mask) { in miopen_convolution_backward()
1447 bool benchmark, bool deterministic, std::array<bool,3> output_mask) { in miopen_depthwise_convolution_backward()
H A DRNN_miopen.cpp44 const Tensor& reserve, std::array<bool, 4> output_mask in miopen_rnn_backward()
565 std::array<bool, 3> output_mask in miopen_rnn_backward_input()
764 const Tensor& reserve, std::array<bool, 4> output_mask in miopen_rnn_backward()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DLinear.cpp44 const Tensor& weight, std::array<bool,3> output_mask) { in mkldnn_linear_backward()
172 const Tensor& weight, std::array<bool,3> output_mask) { in mkldnn_linear_backward()
H A DConv.cpp867 … padding, IntArrayRef stride, IntArrayRef dilation, int64_t groups, std::array<bool,3> output_mask) in mkldnn_convolution_backward()
1019 std::array<bool,3> output_mask) in mkldnn_convolution_transpose_backward()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DConvPlaceholders.cpp96 std::array<bool, 2> output_mask) { in cudnn_convolution_backward()
155 std::array<bool, 2> output_mask) { in cudnn_convolution_transpose_backward()
H A DConvShared.cpp592 std::array<bool, 2> output_mask) { in cudnn_convolution_backward()
672 std::array<bool, 2> output_mask) { in cudnn_convolution_transpose_backward()
H A DRNN.cpp92 std::array<bool, 4> output_mask) { in _cudnn_rnn_backward()
1689 std::array<bool, 3> output_mask) { in _cudnn_rnn_backward_input()
2110 std::array<bool, 4> output_mask) { in _cudnn_rnn_backward()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dshape_inference.cpp271 ::std::array<bool, 3> output_mask) { in compute_shape_convolution_backward()
580 ::std::array<bool, 3> output_mask) { in compute_shape_native_batch_norm_backward()
636 ::std::array<bool, 3> output_mask) { in compute_shape_native_layer_norm_backward()
882 ::std::array<bool, 2> output_mask) { in compute_shape_grid_sampler_2d_backward()
/aosp_15_r20/external/executorch/kernels/test/
H A Dop_convolution_backward_test.cpp44 ArrayRef<bool> output_mask(output_mask_a.data(), output_mask_a.size()); in op_convolution_backward_out() local
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dmaia_extension.cpp47 int64_t groups, std::array<bool,3> output_mask) { in fake_convolution_backward()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorBackward.cpp43 std::array<bool, 3> output_mask) { in nested_linear_backward()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp3290 std::array<bool, 2> output_mask) { in atan2_backward()
4367 std::array<bool, 2> output_mask) { in triangular_solve_backward()
4442 std::array<bool, 2> output_mask) { in linalg_solve_triangular_backward()
4499 std::array<bool, 2> output_mask) { in cholesky_solve_backward()
4663 std::array<bool, 3> output_mask) { in batchnorm_double_backward()
4799 std::array<bool, 3> output_mask) { in layer_norm_double_backward()
6754 ::std::array<bool, 2> output_mask) { in _cudnn_convolution_backward()
/aosp_15_r20/external/libaom/av1/encoder/x86/
H A Dcnn_avx2.c122 static inline void load_shuffle_masks_for_2x2_convolve(__m256i *output_mask, in load_shuffle_masks_for_2x2_convolve()

12