Searched refs:get_activations (Results 1 – 2 of 2) sorted by relevance
715 def get_activations(): function721 sparse_activations = strategy.run(get_activations)723 ragged_activations = strategy.run(get_activations)745 def get_activations(dense_value): function750 activations, dense_value1 = strategy.run(get_activations, args=(0.0,))764 per_device_activations, dense_value2 = strategy.run(get_activations,785 def get_activations(): function790 activations = strategy.run(get_activations)808 def get_activations(features): function812 activations = strategy.run(get_activations, args=(next(sparse_iter),))[all …]
1845 def get_activations(self): member in TPUEmbedding