Searched refs:wt_init (Results 1 – 1 of 1) sorted by relevance
369 def _create_weights(cls, wt_init, num_wts, num_shards, name): argument387 if wt_init is None:391 if isinstance(wt_init, collections.Iterable):392 if num_shards == 1 and len(wt_init) == num_wts:393 wt_init = [wt_init]394 assert len(wt_init) == num_shards395 elif isinstance(wt_init, numbers.Real) and wt_init >= 0:411 return wt_init * array_ops.ones([size])413 return wt_init[i]