Searched refs:always_wrap (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_utils.py | 35 def regroup(values, wrap_class=values_lib.PerReplica, always_wrap=False): argument
54 regroup(tuple(v[i] for v in values), wrap_class, always_wrap)
63 regroup(tuple(v[i] for v in values), wrap_class, always_wrap)
82 wrap_class, always_wrap)
107 if same_id and not always_wrap and not hasattr(v0, "_distributed_container"):
|
| D | input_lib.py | 2187 always_wrap = _always_wrap(strategy)
2188 per_replicas = distribute_utils.regroup(value_list, always_wrap=always_wrap)
|
| D | parameter_server_strategy.py | 407 return distribute_utils.regroup(per_replica_values, always_wrap=True)
|
| D | collective_all_reduce_strategy.py | 633 return distribute_utils.regroup(per_replica_values, always_wrap=True)
|
| D | mirrored_strategy.py | 597 return distribute_utils.regroup(per_replica_values, always_wrap=True)
|
| D | tpu_strategy.py | 985 return distribute_utils.regroup(per_replica_values, always_wrap=True)
|
| /external/libaom/libaom/ |
| D | .cmake-format.py | 52 always_wrap = [] variable
|