Searched refs:wrap_class (Results 1 – 3 of 3) 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)
127 return wrap_class(values)
|
| D | cross_device_ops.py | 197 wrap_class=value_lib.Mirrored)
296 return distribute_utils.regroup((v,), wrap_class=value_lib.Mirrored)
345 return distribute_utils.regroup((v,), wrap_class=value_lib.Mirrored)
421 distribute_utils.regroup(v.values, wrap_class=value_lib.Mirrored)
686 v, wrap_class=value_lib.Mirrored) for v in index]
1160 return distribute_utils.regroup(index, wrap_class=value_lib.Mirrored)
1262 distribute_utils.regroup(values, wrap_class=value_lib.Mirrored))
1300 distribute_utils.regroup(value, wrap_class=value_lib.Mirrored))
1330 return distribute_utils.regroup(index, wrap_class=value_lib.Mirrored)
1370 distribute_utils.regroup(value, wrap_class=value_lib.Mirrored))
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 96 wrap_class=value_lib.Mirrored)
113 wrap_class=value_lib.Mirrored)
|