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)
