Searched refs:gx_mult (Results 1 – 1 of 1) sorted by relevance
595 auto [x, gx_mult] = compute_W.apply_get_grad(grid_x); in backward()688 gx = gx * gx_mult; in backward()959 Vec gx_mult = Vec(compute_W.scaling_factor); in backward() local1004 gx = gx * gx_mult; in backward()