| /external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| D | cuSPARSELtOps.cpp | 104 const std::optional<Tensor>& bias_opt, in _cslt_sparse_mm_impl()
346 const std::optional<Tensor>& bias_opt, in _cslt_sparse_mm()
368 const std::optional<Tensor>& bias_opt, in _cslt_sparse_mm_search()
401 const std::optional<Tensor>& bias_opt, in _cslt_sparse_mm()
413 const std::optional<Tensor>& bias_opt, in _cslt_sparse_mm_search()
|
| D | SparseSemiStructuredLinear.cu | 603 const Tensor& meta, const std::optional<Tensor>& bias_opt, in _sparse_semi_structured_linear()
|
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | Normalization.cpp | 25 …st Tensor& self, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, c… in mkldnn_batch_norm()
49 …t Tensor& input, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, T… in _mkldnn_batch_norm_legit()
58 …onst Tensor& input, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, in _mkldnn_batch_norm_legit_no_stats()
66 …onst Tensor& input, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, in _batch_norm_with_update_mkldnn()
134 …t Tensor& input, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, c… in mkldnn_batch_norm()
212 …onst Tensor& input, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, in _batch_norm_with_update_mkldnn()
222 …t Tensor& input, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, T… in _mkldnn_batch_norm_legit()
231 …onst Tensor& input, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, in _mkldnn_batch_norm_legit_no_stats()
|
| D | Linear.cpp | 29 const Tensor& weight, const std::optional<Tensor>& bias_opt) { in mkldnn_linear()
61 const Tensor& weight_t, const std::optional<Tensor>& bias_opt) { in mkldnn_linear()
186 const std::optional<Tensor>& bias_opt, in mkldnn_linear_pointwise()
261 const std::optional<Tensor>& bias_opt, in mkldnn_linear_pointwise_binary()
345 const std::optional<Tensor>& bias_opt, in mkl_linear()
|
| D | Conv.cpp | 25 const Tensor& input, const Tensor& weight, const std::optional<Tensor>& bias_opt, in mkldnn_convolution()
216 const std::optional<Tensor>& bias_opt, in _mkldnn_convolution()
281 const std::optional<Tensor>& bias_opt, in mkldnn_convolution()
302 const std::optional<Tensor>& bias_opt, in mkldnn_convolution_pointwise()
339 const std::optional<Tensor>& bias_opt, in mkldnn_convolution_pointwise_binary()
503 const std::optional<Tensor>& bias_opt, in mkldnn_convolution_pointwise_binary_()
621 const std::optional<Tensor>& bias_opt, in _mkldnn_convolution_transpose()
717 const std::optional<Tensor>& bias_opt, in mkldnn_convolution_transpose_pointwise()
748 const std::optional<Tensor>& bias_opt, in mkldnn_convolution_transpose_pointwise_meta()
898 const std::optional<Tensor>& bias_opt, in mkldnn_convolution_transpose()
|
| D | ConvPrepack.cpp | 150 const std::optional<Tensor>& bias_opt, in mkldnn_convolution_out()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | layer_norm.cpp | 77 …, const std::optional<Tensor>& weight_opt /* optional */, const std::optional<Tensor>& bias_opt /*… in layer_norm_cpu()
119 const std::optional<Tensor>& bias_opt /* optional */, in layer_norm_backward_cpu()
189 …, const std::optional<Tensor>& weight_opt /* optional */, const std::optional<Tensor>& bias_opt /*… in layer_norm_symint()
207 …f normalized_shape, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, in math_native_layer_norm()
275 auto bias_opt = std::optional<Tensor>(); in rms_norm() local
|
| D | Normalization.cpp | 541 …, const std::optional<Tensor>& weight_opt /* optional */, const std::optional<Tensor>& bias_opt /*… in _batch_norm_impl_index()
669 …onst Tensor& input, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, in batch_norm()
705 …, const std::optional<Tensor>& weight_opt /* optional */, const std::optional<Tensor>& bias_opt /*… in instance_norm()
761 …st Tensor& self, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, c… in batch_norm_cpu_out()
804 …st Tensor& self, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, c… in batch_norm_cpu()
854 …onst Tensor& input, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, in _batch_norm_with_update_cpu()
863 …onst Tensor& input, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, in _batch_norm_with_update_cpu_out()
873 …onst Tensor& input, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, in _batch_norm_no_update()
885 …const Tensor& self, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, in _batch_norm_legit_cpu()
891 …const Tensor& self, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, in _batch_norm_legit_no_stats_cpu()
[all …]
|
| D | NNPACK.cpp | 25 const Tensor& weight, const std::optional<Tensor>& bias_opt, in _nnpack_spatial_convolution()
140 const Tensor& weight, const std::optional<Tensor>& bias_opt, in _nnpack_spatial_convolution()
|
| D | Convolution.cpp | 886 const std::optional<at::Tensor>& bias_opt, in complex_convolution_mode()
916 const Tensor& input_, const Tensor& weight, const std::optional<Tensor>& bias_opt, in conv1d_symint()
941 const Tensor& input_, const Tensor& weight, const std::optional<Tensor>& bias_opt, in conv2d_symint()
966 const Tensor& input_, const Tensor& weight, const std::optional<Tensor>& bias_opt, in conv3d_symint()
1057 const Tensor& input, const Tensor& weight, const std::optional<Tensor>& bias_opt, in _convolution_mode_symint()
1117 const Tensor& input_, const Tensor& weight, const std::optional<Tensor>& bias_opt, in conv_transpose1d_symint()
1136 const Tensor& input_, const Tensor& weight, const std::optional<Tensor>& bias_opt, in conv_transpose2d_symint()
1155 const Tensor& input_, const Tensor& weight, const std::optional<Tensor>& bias_opt, in conv_transpose3d_symint()
1174 const Tensor& input, const Tensor& weight, const std::optional<Tensor>& bias_opt, in convolution()
1190 const Tensor& input, const Tensor& weight, const std::optional<Tensor>& bias_opt, in convolution_overrideable()
[all …]
|
| D | group_norm.cpp | 181 const std::optional<Tensor>& bias_opt /* optional */, in group_norm()
216 const std::optional<Tensor>& bias_opt, in math_group_norm()
|
| D | ConvolutionMM2d.cpp | 541 IntArrayRef kernel_size, const std::optional<Tensor>& bias_opt, in slow_conv2d_forward_out_cpu()
635 IntArrayRef kernel_size, const std::optional<Tensor>& bias_opt, in slow_conv2d_forward_cpu()
734 …r & self, const Tensor & weight, IntArrayRef kernel_size, const std::optional<Tensor>& bias_opt, I… in thnn_conv2d_out()
742 …r & self, const Tensor & weight, IntArrayRef kernel_size, const std::optional<Tensor>& bias_opt, I… in thnn_conv2d()
|
| D | Linear.cpp | 73 Tensor linear(const Tensor& input, const Tensor& weight, const std::optional<Tensor>& bias_opt) { in linear()
124 Tensor& linear_out(const Tensor& input, const Tensor& weight, const std::optional<Tensor>& bias_opt… in linear_out()
712 …ensor& input1, const Tensor& input2, const Tensor& weight, const std::optional<Tensor>& bias_opt) { in bilinear()
|
| D | ConvolutionMM3d.cpp | 556 IntArrayRef kernel_size, const std::optional<Tensor>& bias_opt, in slow_conv3d_forward_out_cpu()
671 IntArrayRef kernel_size, const std::optional<Tensor>& bias_opt, in slow_conv3d_forward_cpu()
774 IntArrayRef kernel_size, const std::optional<Tensor>& bias_opt, in slow_conv3d_out()
795 IntArrayRef kernel_size, const std::optional<Tensor>& bias_opt, in slow_conv3d()
|
| D | NaiveDilatedConvolution.cpp | 527 IntArrayRef kernel_size, const std::optional<Tensor>& bias_opt, in slow_conv_dilated2d_cpu()
582 IntArrayRef kernel_size, const std::optional<Tensor>& bias_opt, in slow_conv_dilated3d_cpu()
|
| D | NaiveConvolutionTranspose3d.cpp | 822 IntArrayRef kernel_size, const std::optional<Tensor>& bias_opt, in slow_conv_transpose3d_out_cpu()
849 IntArrayRef kernel_size, const std::optional<Tensor>& bias_opt, in slow_conv_transpose3d_cpu()
|
| /external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| D | Layernorm.cpp | 67 const std::optional<Tensor>& bias_opt = in run_layernorm_context() local
85 const std::optional<Tensor>& bias_opt /* optional */, in layer_norm()
|
| D | Batchnorm.cpp | 77 const std::optional<Tensor>& bias_opt /* optional */, in batch_norm()
108 const std::optional<Tensor>& bias_opt, in BatchNormPackedContext()
185 std::optional<Tensor>&& bias_opt, in create_batchnorm_context()
|
| D | NativeLayerNorm.cpp | 59 const std::optional<Tensor>& bias_opt /* optional */, in native_layer_norm()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Normalization.cu | 99 const std::optional<Tensor>& bias_opt, const Tensor& mean_, const Tensor& invstd_) { in batch_norm_elementwise()
435 …st Tensor& self, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, c… in batch_norm_cuda_out()
461 …st Tensor& self, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, c… in batch_norm_cuda()
485 …onst Tensor& input, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, in _batch_norm_with_update_cuda()
510 …onst Tensor& input, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, in _batch_norm_with_update_cuda_out()
532 …st Tensor& self, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, T… in _batch_norm_legit_cuda()
536 …st Tensor& self, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, b… in _batch_norm_legit_no_stats_cuda()
540 …st Tensor& self, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, T… in _batch_norm_legit_cuda_out()
544 …st Tensor& self, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, b… in _batch_norm_legit_no_stats_cuda_out()
677 const std::optional<Tensor>& bias_opt, const Tensor& mean, in batch_norm_elemt_cuda()
[all …]
|
| D | MixedDtypesLinear.cu | 199 const std::optional<Tensor>& bias_opt, in _mixed_dtypes_linear()
|
| D | NaiveDilatedConvolution.cu | 402 IntArrayRef kernel_size, const std::optional<Tensor>& bias_opt, in slow_conv_dilated2d_cuda()
508 IntArrayRef kernel_size, const std::optional<Tensor>& bias_opt, in slow_conv_dilated3d_cuda()
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesNorm.cpp | 49 const std::optional<Tensor>& bias_opt, std::optional<int64_t> bias_bdim, in batch_norm_batch_rule()
288 const std::optional<Tensor> & bias_opt, int64_t N, int64_t C, in native_group_norm_plumbing()
492 const std::optional<Tensor>& bias_opt, std::optional<int64_t> bias_bdim, in native_layer_norm_batch_rule()
577 const std::optional<at::Tensor> & bias_opt, in native_layer_norm_backward_plumbing()
853 …const Tensor& self, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, in _native_batch_norm_legit_batch()
859 …const Tensor& self, const std::optional<Tensor>& weight_opt, const std::optional<Tensor>& bias_opt, in _native_batch_norm_legit_no_stats_batch()
|
| D | PyTorchOperatorHacks.cpp | 36 Tensor linear_hack(const Tensor& input, const Tensor& weight, const std::optional<Tensor>& bias_opt… in linear_hack()
|
| /external/pytorch/aten/src/ATen/native/miopen/ |
| D | BatchNorm_miopen.cpp | 25 …const Tensor& input, const Tensor& weight, const std::optional<Tensor>& bias_opt, const std::optio… in miopen_batch_norm()
|