def schedule_add_summary(self, stamp_token, column, example_weights): """Schedules to add a quantile summary to its stream in resource.""" summary = self._make_summary(column, example_weights) return batch_ops_utils.ScheduledStampedResourceOp( op=gen_quantile_ops.quantile_accumulator_add_summaries, resource_handle=self.resource_handle, summaries=summary)
def schedule_add(self, partition_ids, feature_ids, gradients, hessians): """Schedules an update to the stats accumulator.""" partition_ids, feature_ids, gradients, hessians = (self._make_summary( partition_ids, feature_ids, gradients, hessians)) if self._is_scalar: return batch_ops_utils.ScheduledStampedResourceOp( op=gen_stats_accumulator_ops.stats_accumulator_scalar_add, resource_handle=self._resource_handle, partition_ids=partition_ids, feature_ids=feature_ids, gradients=gradients, hessians=hessians) else: return batch_ops_utils.ScheduledStampedResourceOp( op=gen_stats_accumulator_ops.stats_accumulator_tensor_add, resource_handle=self._resource_handle, partition_ids=partition_ids, feature_ids=feature_ids, gradients=gradients, hessians=hessians)
def schedule_get_buckets(self): """Returns a scheduled read of buckets created during previous flush.""" return batch_ops_utils.ScheduledStampedResourceOp( resource_handle=self.resource_handle, op=gen_quantile_ops.quantile_accumulator_get_buckets)