Home
last modified time | relevance | path

Searched defs:get_slot (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dtpu_optimizer.py194 def get_slot(self, *args, **kwargs): member in CrossShardOptimizer
/external/tensorflow/tensorflow/python/training/
Dsync_replicas_optimizer.py369 def get_slot(self, *args, **kwargs): member in SyncReplicasOptimizer
Doptimizer.py745 def get_slot(self, var, name): member in Optimizer
/external/mesa3d/src/compiler/nir/
Dnir_lower_io_to_vector.c42 get_slot(const nir_variable *var) in get_slot() function
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2.py921 def get_slot(self, var, slot_name): member in OptimizerV2
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py876 def get_slot(self, var, slot_name): member in LossScaleOptimizer