Searched defs:get_slot (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_optimizer.py | 194 def get_slot(self, *args, **kwargs): member in CrossShardOptimizer
|
| /external/tensorflow/tensorflow/python/training/ |
| D | sync_replicas_optimizer.py | 369 def get_slot(self, *args, **kwargs): member in SyncReplicasOptimizer
|
| D | optimizer.py | 745 def get_slot(self, var, name): member in Optimizer
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_lower_io_to_vector.c | 42 get_slot(const nir_variable *var) in get_slot() function
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 921 def get_slot(self, var, slot_name): member in OptimizerV2
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 876 def get_slot(self, var, slot_name): member in LossScaleOptimizer
|