Searched defs:weight_v (Results 1 – 4 of 4) sorted by relevance
324 def forward(self, weight_g, weight_v): argument
45 float weight_v = weight_data ? weight_data[c] : 1; in compute_fused_params() local
4496 weight_v, argument4529 weight_v, argument4607 weight_v, argument4636 weight_v, argument5773 def _weight_norm(g: jit_utils.GraphContext, weight_v, weight_g, dim): argument
65 param_t weight_v = weight_data ? weight_data[c] : param_t(1); in batch_norm_cpu_collect_linear_and_constant_terms() local