def __init__(self, combiner, serialized_tf_config):
if isinstance(combiner, analyzers.QuantilesCombiner):
tf_config = common._maybe_deserialize_tf_config( # pylint: disable=protected-access
serialized_tf_config)
combiner.initialize_local_state(tf_config)
self._combiner = combiner
self._serialized_tf_config = serialized_tf_config
def __init__(self,
combiner,
serialized_tf_config,
is_combining_accumulators,
should_extract_output=None):
"""Init method for _CombinerWrapper.
Args:
combiner: A `analyzer_nodes.Combiner` object used to combine.
serialized_tf_config: A str which is a serialized form of
`tf.ConfigProto`.
is_combining_accumulators: A bool which indicates whether this is
combining single or batched inputs, or already accumulated objects.
should_extract_output: A bool which indicates whether this should call the
combiner's extract_output method in extract_output. If not specified, we
assume it's the same value as `should_extract_output`.
"""
# TODO(b/69566045): Move initialization to start_bundle(), removing the need
# for initialize_local_state to be called here.
if isinstance(combiner, analyzers.QuantilesCombiner):
tf_config = common._maybe_deserialize_tf_config( # pylint: disable=protected-access
serialized_tf_config)
combiner.initialize_local_state(tf_config)
self._combiner = combiner
self._serialized_tf_config = serialized_tf_config
self._is_combining_accumulators = is_combining_accumulators
if should_extract_output is None:
should_extract_output = is_combining_accumulators
self._should_extract_output = should_extract_output
def _make_graph_state(self, saved_model_dir):
start = datetime.datetime.now()
tf_config = common._maybe_deserialize_tf_config( # pylint: disable=protected-access
self._serialized_tf_config)
result = self._GraphState(saved_model_dir, self._input_schema,
self._exclude_outputs, tf_config)
self._graph_load_seconds_distribution.update(
int((datetime.datetime.now() - start).total_seconds()))
return result
def __init__(self, spec, serialized_tf_config):
if isinstance(spec, analyzers._QuantilesCombinerSpec): # pylint: disable=protected-access
spec.initialize_local_state(
common._maybe_deserialize_tf_config(serialized_tf_config)) # pylint: disable=protected-access
self._spec = spec
self._serialized_tf_config = serialized_tf_config
