Home
last modified time | relevance | path

Searched refs:call_fn (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dutils.py41 def use_wrapped_call(layer, call_fn, default_training_value=None, argument
58 if hasattr(call_fn, 'original_call'): # call_fn is a LayerCall object
59 original_call = call_fn.original_call
61 call_fn = call_fn.__call__
63 original_call = call_fn
65 original_call, call_fn, expects_training_arg, default_training_value)
90 target=call_fn,
203 def get_training_arg_index(call_fn): argument
215 arg_list = tf_inspect.getfullargspec(call_fn).args
216 if tf_inspect.ismethod(call_fn):
Dsave_impl.py165 call_fn = call_collection.add_function(
170 '__call__': call_fn}
456 def _maybe_wrap_with_training_arg(self, call_fn): argument
460 arg_spec = tf_inspect.getfullargspec(call_fn)
475 if tf_inspect.ismethod(call_fn):
485 return call_fn(*args, **kwargs)
488 target=call_fn,
492 return call_fn
494 def add_function(self, call_fn, name): argument
497 self, self._maybe_wrap_with_training_arg(call_fn), name,
[all …]
Dload.py813 call_fn = _get_keras_attr(layer).call_and_return_conditional_losses
814 if call_fn.input_signature is None:
815 inputs = infer_inputs_from_restored_call_function(call_fn)
817 inputs = call_fn.input_signature[0]
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer.py846 call_fn = self.call
855 call_fn = autograph.tf_convert(self.call, ag_ctx.control_status_ctx())
882 outputs = call_fn(inputs, *args, **kwargs)
1020 call_fn = self.call
1024 call_fn = self._autographed_call()
1032 outputs = call_fn(inputs, *args, **kwargs)
1158 call_fn = autograph.tf_convert(self.call,
1161 call_fn = self.call
1166 outputs = call_fn(cast_inputs, *args, **kwargs)
Dbase_layer_v1.py775 call_fn = autograph.tf_convert(
778 call_fn = self.call
784 outputs = call_fn(cast_inputs, *args, **kwargs)
/external/tensorflow/tensorflow/python/keras/
Dlosses.py155 call_fn = self.call
157 call_fn = autograph.tf_convert(self.call, ag_ctx.control_status_ctx())
158 losses = call_fn(y_true, y_pred)