Home
last modified time | relevance | path

Searched defs:bias_opt (Results 1 – 25 of 39) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DNormalization.cpp25 …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()
H A DConv.cpp25 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()
H A DLinear.cpp29 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()
H A DConvPrepack.cpp150 const std::optional<Tensor>& bias_opt, in mkldnn_convolution_out()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNormalization.cpp541 …, 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 …]
H A Dlayer_norm.cpp77 …, 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
H A DConvolution.cpp886 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 …]
H A DConvolutionMM2d.cpp541 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()
H A DNNPACK.cpp25 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()
H A DConvolutionMM3d.cpp556 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()
H A Dgroup_norm.cpp181 const std::optional<Tensor>& bias_opt /* optional */, in group_norm()
216 const std::optional<Tensor>& bias_opt, in math_group_norm()
H A DLinear.cpp73 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()
H A DNaiveDilatedConvolution.cpp527 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()
H A DNaiveConvolutionTranspose3d.cpp822 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()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNormalization.cu99 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 …]
H A DNaiveDilatedConvolution.cu402 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()
H A DMixedDtypesLinear.cu199 const std::optional<Tensor>& bias_opt, in _mixed_dtypes_linear()
H A DNaiveConvolutionTranspose3d.cu838 IntArrayRef kernel_size, const std::optional<Tensor>& bias_opt, in slow_conv_transpose3d_out_cuda()
865 IntArrayRef kernel_size, const std::optional<Tensor>& bias_opt, in slow_conv_transpose3d_cuda()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DcuSPARSELtOps.cpp104 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()
H A DSparseSemiStructuredLinear.cu603 const Tensor& meta, const std::optional<Tensor>& bias_opt, in _sparse_semi_structured_linear()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DBatchnorm.cpp77 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()
H A DLayernorm.cpp67 const std::optional<Tensor>& bias_opt = in run_layernorm_context() local
85 const std::optional<Tensor>& bias_opt /* optional */, in layer_norm()
H A DNativeLayerNorm.cpp59 const std::optional<Tensor>& bias_opt /* optional */, in native_layer_norm()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesNorm.cpp49 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()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DBatchNorm_miopen.cpp25 …const Tensor& input, const Tensor& weight, const std::optional<Tensor>& bias_opt, const std::optio… in miopen_batch_norm()

12