Searched defs:new_w (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesConvolution.cpp | 50 auto new_w = reshape_dim_into(*rhs_bdim, rhs_spec[0], rhs); in convolution_batch_rule() local
63 auto new_w = reshape_dim_into(*rhs_bdim, 1, rhs); in convolution_batch_rule() local
81 auto new_w = reshape_dim_outof_symint(0 + (*rhs_bdim == 0), groups, rhs); in convolution_batch_rule() local
101 auto new_w = reshape_dim_into(*rhs_bdim, rhs_spec[dim_with_groups], rhs); in convolution_batch_rule() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | fold_conv_bn.cpp | 34 at::Tensor new_w = p.conv_w * (p.bn_w * bn_var_rsqrt).reshape(sizes); in computeUpdatedConvWeightAndBias() local
|
| /aosp_15_r20/external/skia/src/core/ |
| H A D | SkBlurEngine.cpp | 1016 auto get_new_weight = [](float* new_w, float* offset, float wi, float wj) { in Compute1DBlurLinearKernel()
|