Home
last modified time | relevance | path

Searched refs:PerReplica (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dtest_util.py58 value = values.PerReplica([ops.convert_to_tensor(value)])
64 return strategy.gather(values.PerReplica(inputs), axis=0)
Dvalues_test.py605 value = values_lib.PerReplica([1., 2.])
1388 per_replica = values_lib.PerReplica(vals)
1397 per_replica = values_lib.PerReplica(vals)
1408 per_replica = values_lib.PerReplica(vals)
1421 per_replica = values_lib.PerReplica((constant_op.constant(1.),))
1434 per_replica = values_lib.PerReplica((constant_op.constant(1.),))
1448 self.assertIsInstance(output, values_lib.PerReplica)
1455 x = values_lib.PerReplica((constant_op.constant(1.),))
1463 per_replica_1 = values_lib.PerReplica((constant_op.constant("a"),))
1464 per_replica_2 = values_lib.PerReplica((constant_op.constant(["b", "c"]),))
[all …]
Ddistribute_utils_test.py45 def _is_per_replica(self, result, expected, klass=values.PerReplica):
146 self._is_per_replica(result[0], ("1", "3"), values.PerReplica)
147 self._is_per_replica(result[1], ("2", "4"), values.PerReplica)
Dvars_test.py178 per_replica_value = values.PerReplica(
181 per_replica_sub_value = values.PerReplica(
761 per_replica_value = values.PerReplica(
1156 delta = values.PerReplica([
1174 delta = values.PerReplica([
1192 delta = values.PerReplica([
1210 delta = values.PerReplica([
1228 delta = values.PerReplica([
1246 delta = values.PerReplica([
1264 delta = values.PerReplica([
Dcross_device_ops.py118 if isinstance(input_tensor, value_lib.PerReplica):
121 return value_lib.PerReplica((input_tensor,))
155 if not all(isinstance(v[0], value_lib.PerReplica)
Dvalues.py61 not var.dtype.is_floating) and isinstance(value, PerReplica):
362 class PerReplica(DistributedValues, composite_tensor.CompositeTensor): class
381 value_type = property(lambda self: PerReplica)
402 return PerReplica(tensor_list)
Ddistribute_utils.py35 def regroup(values, wrap_class=values_lib.PerReplica, always_wrap=False):
Dps_values.py118 if name and isinstance(name, values.PerReplica):
Dparameter_server_strategy.py542 elif isinstance(x, values.PerReplica):
Dmirrored_variable_test.py434 names = values.PerReplica(("foo", "bar"))
Dmirrored_strategy_test.py1139 factors = values.PerReplica((5.0, 3.0))
1140 expected_result = values.PerReplica((5.0 * 1.25, 3.0 * 1.25))
Dcross_device_ops_test.py92 return value_lib.PerReplica(values)
Dtpu_strategy.py1624 last_step_tensor_outputs_dict[name] = values.PerReplica(output)
Dinput_lib.py571 worker_has_values = values.PerReplica(worker_has_values)
/external/tensorflow/tensorflow/python/keras/distribute/
Dcustom_training_loop_optimizer_test.py58 grads = values.PerReplica([
/external/tensorflow/tensorflow/python/distribute/v1/
Dcross_device_ops_test.py243 per_replica = value_lib.PerReplica((t0, t1))
372 per_replica = value_lib.PerReplica((t0, t1))
662 per_replica = value_lib.PerReplica(indexed_slices)
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining.py2753 if not isinstance(v, ds_values.PerReplica):
2807 if isinstance(v, ds_values.PerReplica):
/external/tensorflow/tensorflow/python/ops/
Dstateful_random_ops_test.py676 gens = dist_values.PerReplica([[g] for g in gens])
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2.py710 if isinstance(grad, ds_values.PerReplica):