Home
last modified time | relevance | path

Searched defs:weight_v (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/nn/utils/
Dparametrizations.py324 def forward(self, weight_g, weight_v): argument
/external/pytorch/aten/src/ATen/native/quantized/cpu/
DNormalization.cpp45 float weight_v = weight_data ? weight_data[c] : 1; in compute_fused_params() local
/external/pytorch/torch/onnx/
Dsymbolic_opset9.py4496 weight_v, argument
4529 weight_v, argument
4607 weight_v, argument
4636 weight_v, argument
5773 def _weight_norm(g: jit_utils.GraphContext, weight_v, weight_g, dim): argument
/external/pytorch/aten/src/ATen/native/cpu/
Dbatch_norm_kernel.cpp65 param_t weight_v = weight_data ? weight_data[c] : param_t(1); in batch_norm_cpu_collect_linear_and_constant_terms() local