| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | GridSampler.cpp | 47 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()
|
| D | NaiveDilatedConvolution.cu | 458 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()
|
| D | DepthwiseConv3d.cu | 538 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()
|
| D | ConvolutionMM2d.cu | 473 std::array<bool, 3> output_mask) { in slow_conv2d_backward_cuda()
|
| D | NaiveConvolutionTranspose2d.cu | 768 std::array<bool, 3> output_mask) { in slow_conv_transpose2d_backward_cuda()
|
| D | GridSampler.cu | 843 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()
|
| /external/executorch/kernels/portable/cpu/ |
| D | op_convolution_backward.cpp | 37 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()
|
| /external/pytorch/test/ |
| D | test_meta.py | 1383 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
|
| /external/executorch/kernels/test/ |
| D | op_convolution_backward_test.cpp | 44 ArrayRef<bool> output_mask(output_mask_a.data(), output_mask_a.size()); in op_convolution_backward_out() local
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesNorm.cpp | 209 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()
|
| D | BatchRulesConvolution.cpp | 142 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()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | NaiveDilatedConvolution.cpp | 638 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()
|
| D | Convolution.cpp | 1724 … 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()
|
| D | GridSampler.cpp | 209 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()
|
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | Linear.cpp | 44 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()
|
| D | Conv.cpp | 867 … 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()
|
| /external/pytorch/aten/src/ATen/native/cudnn/ |
| D | ConvPlaceholders.cpp | 96 std::array<bool, 2> output_mask) { in cudnn_convolution_backward()
155 std::array<bool, 2> output_mask) { in cudnn_convolution_transpose_backward()
|
| D | ConvShared.cpp | 592 std::array<bool, 2> output_mask) { in cudnn_convolution_backward()
672 std::array<bool, 2> output_mask) { in cudnn_convolution_transpose_backward()
|
| /external/pytorch/test/cpp_extensions/ |
| D | maia_extension.cpp | 47 int64_t groups, std::array<bool,3> output_mask) { in fake_convolution_backward()
|
| /external/pytorch/torch/csrc/lazy/core/ |
| D | shape_inference.cpp | 271 ::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()
|
| /external/pytorch/aten/src/ATen/native/miopen/ |
| D | Conv_miopen.cpp | 62 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()
|
| D | RNN_miopen.cpp | 44 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()
|
| /external/pytorch/aten/src/ATen/native/nested/ |
| D | NestedTensorBackward.cpp | 43 std::array<bool, 3> output_mask) { in nested_linear_backward()
|
| /external/libaom/av1/encoder/x86/ |
| D | cnn_avx2.c | 122 static inline void load_shuffle_masks_for_2x2_convolve(__m256i *output_mask, in load_shuffle_masks_for_2x2_convolve()
|
| /external/pytorch/aten/src/ATen/native/mkldnn/xpu/ |
| D | Conv.cpp | 608 std::array<bool, 3> output_mask) { in convolution_backward_overrideable()
|