def _make_saveable(self, name): bucket_boundaries = get_bucket_boundaries(self._resource_handle, self._num_streams) slice_spec = '' specs = [] for i in range(self._num_streams): specs.append( saver.BaseSaverBuilder.SaveSpec( bucket_boundaries[i], slice_spec, name + '_bucket_boundaries_' + str(i))) super(QuantileAccumulator, self).__init__(self._resource_handle, specs, name) ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, self)
def __init__(self, resource_handle, create_op, num_streams, name): self._resource_handle = resource_handle self._num_streams = num_streams self._create_op = create_op bucket_boundaries = get_bucket_boundaries(self._resource_handle, self._num_streams) slice_spec = '' specs = [] def make_save_spec(tensor, suffix): return saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name + suffix) for i in range(self._num_streams): specs += [ make_save_spec(bucket_boundaries[i], '_bucket_boundaries_' + str(i)) ] super(QuantileAccumulatorSaveable, self).__init__(self._resource_handle, specs, name)
def get_bucket_boundaries(self): return get_bucket_boundaries(self.resource_handle, self._num_streams)