| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | scatter_ops_test.py | 111 state_ops.scatter_max: _NumpyMax,
121 state_ops.scatter_max: _NumpyMaxScalar,
219 self._VariableRankTests(state_ops.scatter_max, False)
237 self._VariableRankTests(state_ops.scatter_max, True)
258 self._VariableRankTests(state_ops.scatter_max, False, True)
276 self._VariableRankTests(state_ops.scatter_max, True, True)
|
| D | scatter_nd_ops_test.py | 396 max_result = state_ops.scatter_max(ref, indices, updates)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_values.py | 207 def scatter_max(self, sparse_delta, use_locking=False, name=None): member in TPUDistributedVariable
209 return self._primary.scatter_max(sparse_delta, use_locking, name)
210 return self._policy.scatter_max(
321 def scatter_max(self, *args, **kwargs): member in TPUMirroredVariable
323 return self._primary.scatter_max(*args, **kwargs)
502 def scatter_max(self, var, sparse_delta, use_locking=False, name=None): member in TPUOnWritePolicy
573 def scatter_max(self, *args, **kwargs): member in TPUOnReadPolicy
|
| D | packed_distributed_variable.py | 214 def scatter_max(self, sparse_delta, use_locking=False, name=None): member in PackedDistributedVariable
215 scatter_max_fn = lambda var, *a, **kw: var.scatter_max(*a, **kw)
338 def scatter_max(self, sparse_delta, use_locking=False, name=None): member in PackedVarAndDevice
340 return self._var.scatter_max(sparse_delta, use_locking, name)
|
| D | values.py | 852 def scatter_max(self, sparse_delta, use_locking=False, name=None): member in DistributedVariable
854 return self._primary.scatter_max(sparse_delta, use_locking, name)
856 return self._policy.scatter_max(
858 return values_util.scatter_max(
1110 def scatter_max(self, *args, **kwargs): member in MirroredVariable
1112 return self._primary.scatter_max(*args, **kwargs)
1118 return super(MirroredVariable, self).scatter_max(*args, **kwargs)
1297 def scatter_max(self, *args, **kwargs): member in SyncOnReadVariable
1299 return self._primary.scatter_max(*args, **kwargs)
1594 def scatter_max(self, *args, **kwargs): member in OnReadPolicy
[all …]
|
| D | values_v2.py | 191 def scatter_max(self, sparse_delta, use_locking=False, name=None): member in DistributedVariable
193 return super().scatter_max(sparse_delta, use_locking, name)
|
| D | values_util.py | 266 def scatter_max(var, sparse_delta, use_locking=False, name=None): function
267 scatter_max_fn = lambda var, *a, **kw: var.scatter_max(*a, **kw)
|
| D | vars_test.py | 609 def scatter_max(v): function
614 return v.scatter_max(value)
619 distribution.run(scatter_max, args=(v1,))))
623 distribution.run(scatter_max, args=(v2,))))
1313 self.evaluate(distribution.run(v.scatter_max, args=(delta,)))
|
| D | sharded_variable.py | 633 def scatter_max(self, sparse_delta, use_locking=False, name=None): member in ShardedVariableMixin
640 v.scatter_max(per_var_sparse_delta[i], name=new_name)
|
| D | distributed_variable_test.py | 411 w.scatter_max(_make_index_slices(values=[1., 0.5], indices=[1, 2])),
542 _assert_unsaveable(lambda: v.scatter_max(_make_index_slices([1.], [0])))
|
| D | values_v2_test.py | 156 v.scatter_max(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | state_ops.py | 836 def scatter_max(ref, indices, updates, use_locking=False, name=None): function
883 return gen_state_ops.scatter_max(ref, indices, updates,
|
| D | standard_ops.py | 91 from tensorflow.python.ops.state_ops import scatter_max
|
| D | variables.py | 666 def scatter_max(self, sparse_delta, use_locking=False, name=None): member in Variable
2148 def scatter_max(self, sparse_delta, use_locking=False, name=None): member in RefVariable
2166 return gen_state_ops.scatter_max(
|
| D | resource_variable_ops.py | 1008 def scatter_max(self, sparse_delta, use_locking=False, name=None): member in BaseResourceVariable
2196 def scatter_max(self, sparse_delta, use_locking=False, name=None): member in _UnreadVariable
2198 return super(_UnreadVariable, self).scatter_max(sparse_delta, use_locking,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.__internal__.distribute.-sharded-variable.pbtxt | 61 name: "scatter_max"
|
| D | tensorflow.-variable.pbtxt | 127 name: "scatter_max"
|
| D | tensorflow.experimental.dtensor.-d-variable.pbtxt | 156 name: "scatter_max"
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | autocast_variable.py | 267 def scatter_max(self, sparse_delta, use_locking=False, name=None): member in AutoCastVariable
268 return self._apply_update(self._variable.scatter_max, sparse_delta,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-variable.pbtxt | 128 name: "scatter_max"
|
| D | tensorflow.pbtxt | 2016 name: "scatter_max"
|
| /external/tensorflow/tensorflow/python/kernel_tests/variables/ |
| D | resource_variable_ops_test.py | 621 v.scatter_max(
631 v.scatter_max(
1083 assert_eq(var.scatter_max(slices2).scatter_add(slices), [3., 4.])
|
| /external/tensorflow/ |
| D | RELEASE.md | 8273 * Add `tf.scatter_min` and `tf.scatter_max`
|