Searched refs:all_params (Results 1 – 5 of 5) sorted by relevance
251 all_params = [271 xs = all_params278 f_restricted, [all_params[param_index]], delta=1e-5))
257 all_params = [283 all_params[param_index],
150 all_params = [x, m, v, beta, gamma]152 err = gradient_checker.compute_gradient_error(all_params[param_index],
223 all_params = set(params)224 if len(all_params) < len(params):227 if t in all_params:229 all_params.remove(t)231 assert not all_params, all_params
314 const HloVec all_params( in TEST_F() local319 for (const HloInstruction* param : all_params) { in TEST_F()320 for (const HloInstruction* param2 : all_params) { in TEST_F()