Lines Matching refs:gradients
51 gradient_shape: A TensorShape, containing shape of gradients.
71 def update_stats(self, stamp_token, example_partition_ids, gradients,
80 gradients: A dense tensor of gradients.
83 0) as gradients.
96 def update_stats_sync(self, stamp_token, example_partition_ids, gradients,
105 gradients: A dense tensor of gradients.
108 0) as gradients.
122 stamp_token, example_partition_ids, gradients, hessians,