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
|