| /external/tensorflow/tensorflow/lite/python/ |
| D | tflite_keras_util.py | 151 def trace_model_call(model, input_signature=None): argument
165 if input_signature is None:
167 input_signature = model.call.input_signature
169 if input_signature is None:
170 input_signature = model_input_signature(model)
172 if input_signature is None:
175 @def_function.function(input_signature=input_signature, autograph=False)
180 inputs = args[0] if len(input_signature) == 1 else list(args)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/ |
| D | structured_output.py | 40 @tf.function(input_signature=[])
52 @tf.function(input_signature=[])
65 @tf.function(input_signature=[])
78 @tf.function(input_signature=[])
88 @tf.function(input_signature=[])
101 @tf.function(input_signature=[])
116 @tf.function(input_signature=[tf.TensorSpec([], tf.float32)])
|
| D | structured_input.py | 42 @tf.function(input_signature=[
55 @tf.function(input_signature=[[
69 @tf.function(input_signature=[{
83 @tf.function(input_signature=[{
100 @tf.function(input_signature=[{
119 @tf.function(input_signature=[{
|
| D | call_to_exported.py | 68 @tf.function(input_signature=[tf.TensorSpec([], tf.float32)])
72 @tf.function(input_signature=[tf.TensorSpec([], tf.float32)])
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | map_defun_op_test.py | 53 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
68 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
81 @function.defun(input_signature=[tensor_spec.TensorSpec([], dtypes.int32)])
95 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
108 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
122 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
134 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
145 @function.defun(input_signature=[
165 @function.defun(input_signature=[tensor_spec.TensorSpec([], dtypes.int32)])
178 @function.defun(input_signature=[tensor_spec.TensorSpec([5], dtypes.int64)])
[all …]
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saving_utils.py | 101 def trace_model_call(model, input_signature=None): argument
115 if input_signature is None:
117 input_signature = model.call.input_signature
119 if input_signature is None:
120 input_signature = model_input_signature(model)
122 if input_signature is None:
125 @def_function.function(input_signature=input_signature)
130 inputs = args[0] if len(input_signature) == 1 else list(args)
|
| D | saved_model_experimental.py | 69 input_signature=None, argument
137 signatures=saving_utils.trace_model_call(model, input_signature))
140 input_signature)
167 def _save_v1_format(model, path, custom_objects, as_text, input_signature): argument
205 'input_signature': input_signature}
237 input_signature): argument
265 if input_signature is None:
268 input_tensors = nest.map_structure(create_placeholder, input_signature)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | functional_ops_test.py | 51 input_signature=[tensor_spec.TensorSpec((), dtypes.float32)])
56 input_signature=[tensor_spec.TensorSpec((), dtypes.float32)])
69 @def_function.function(input_signature=signature)
73 @def_function.function(input_signature=signature)
|
| D | sobol_ops_test.py | 92 input_signature=[tensor_spec.TensorSpec(shape=[], dtype=dtypes.int32)])
103 input_signature=[tensor_spec.TensorSpec(shape=[], dtype=dtypes.int32)])
114 input_signature=[tensor_spec.TensorSpec(shape=[], dtype=dtypes.int32)] *
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | save_test.py | 95 input_signature=[tensor_spec.TensorSpec(None, dtypes.float32)])
113 input_signature=[tensor_spec.TensorSpec(None, dtypes.float32)])
171 @def_function.function(input_signature=[])
176 @def_function.function(input_signature=[])
188 @def_function.function(input_signature=[])
239 input_signature=([
280 model.f = def_function.function(lambda: 3., input_signature=())
320 @def_function.function(input_signature=[
336 @def_function.function(input_signature=[
358 @def_function.function(input_signature=[
[all …]
|
| D | load_test.py | 147 input_signature=[tensor_spec.TensorSpec([None], dtypes.int64)])
165 input_signature=[tensor_spec.TensorSpec(None, dtypes.float32)])
179 input_signature=[tensor_spec.TensorSpec(None, dtypes.float32)])
252 @def_function.function(input_signature=[
267 input_signature=[])
280 input_signature=[])
327 input_signature=[tensor_spec.TensorSpec(None, dtypes.float32)])
356 input_signature=[tensor_spec.TensorSpec(None, dtypes.float32)])
359 input_signature=[tensor_spec.TensorSpec(None, dtypes.float32)])
668 input_signature=[tensor_spec.TensorSpec(None, dtypes.float32)])
[all …]
|
| D | function_serialization.py | 47 proto.input_signature.CopyFrom(
48 coder.encode_structure(function_spec.input_signature))
|
| /external/tensorflow/tensorflow/python/distribute/integration_test/ |
| D | saved_model_test.py | 90 @tf.function(input_signature=[])
121 @tf.function(input_signature=[])
147 @tf.function(input_signature=[])
173 @tf.function(input_signature=[])
199 @tf.function(input_signature=[])
229 @tf.function(input_signature=[])
238 @tf.function(input_signature=[])
292 @tf.function(input_signature=[])
345 @tf.function(input_signature=[tf.TensorSpec(shape=(), dtype=tf.float32)])
384 @tf.function(input_signature=[])
[all …]
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | def_function.py | 487 input_signature=None, argument
515 input_signature,
536 self._input_signature = input_signature
662 input_signature=self.input_signature,
733 input_signature=self._input_signature,
770 self._python_function, self.input_signature)
1051 def input_signature(self): member in Function
1052 return self._function_spec.input_signature
1144 if self.input_signature is not None:
1337 input_signature=None, argument
[all …]
|
| D | function.py | 1629 function_spec.input_signature,
2339 input_signature, argument
2449 input_signature,
2458 input_signature, argument
2506 if input_signature is None:
2514 if not isinstance(input_signature, (tuple, list)):
2516 "list, received " + str(type(input_signature)))
2518 self._input_signature = tuple(input_signature)
2519 self._flat_input_signature = tuple(nest.flatten(input_signature,
2539 def input_signature(self): member in FunctionSpec
[all …]
|
| D | def_function_test.py | 300 input_signature=[tensor_spec.TensorSpec((), dtypes.int32)])
306 input_signature=[tensor_spec.TensorSpec(None, dtypes.float32)])
677 def testClone(self, input_signature, autograph, autograph_options, implements, argument
684 input_signature=input_signature,
700 self.assertEqual(input_signature, cloned._input_signature)
759 def test_pickle(self, input_signature, autograph, autograph_options, argument
766 input_signature=input_signature,
776 self.assertEqual(input_signature, cloned._input_signature)
|
| /external/tensorflow/tensorflow/cc/saved_model/testdata/ |
| D | generate_saved_models.py | 51 @def_function.function(input_signature=[
81 @def_function.function(input_signature=[])
101 input_signature=[tensor_spec.TensorSpec(shape=None, dtype=dtypes.string)])
|
| /external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| D | save_impl.py | 191 if fn is not None and fn.input_signature is not None:
377 layer.call.input_signature is not None):
378 return layer.call.input_signature
393 input_signature = [nest.map_structure(
396 return input_signature
498 input_signature=self.fn_input_signature)
619 fn, 'loss_fn_{}'.format(index), input_signature=[])
630 input_signature=[
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | prefetching_ops.py | 134 @function.defun(input_signature=[tensor_spec.TensorSpec([], dtypes.string)])
154 input_signature=[tensor_spec.TensorSpec([], dtypes.string)],
166 @function.defun(input_signature=[tensor_spec.TensorSpec([], dtypes.string)])
185 @function.defun(input_signature=[tensor_spec.TensorSpec([], dtypes.string)])
|
| /external/tensorflow/tensorflow/lite/examples/experimental_new_converter/ |
| D | stack_trace_example.py | 49 @tf.function(input_signature=[tf.TensorSpec(shape=[3, 3], dtype=tf.float32)])
73 @tf.function(input_signature=[tf.TensorSpec(shape=[3, 3], dtype=tf.float32)])
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| D | ngrams_test.py | 65 input_signature = [tf.TensorSpec(shape=[None], dtype=tf.string)]
67 input_signature += [tf.TensorSpec(shape=[None], dtype=tf.int64)] * rs
71 @tf.function(input_signature=input_signature)
91 input_signature=[tf.TensorSpec(shape=shape, dtype=tf.string)])
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | make_aot_compile_models.py | 46 input_signature=[tensor_spec.TensorSpec([3000, 5000], dtypes.float32),
63 input_signature=[tensor_spec.TensorSpec([3, 5], dtypes.float32),
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/internal/testdata/ |
| D | gen_saved_models.py | 68 input_signature=[tensor_spec.TensorSpec((), dtypes.float32)])
84 input_signature=[tensor_spec.TensorSpec((), dtypes.float32)])
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | convert_to_constants_test.py | 118 @def_function.function(input_signature=[
140 @def_function.function(input_signature=[
169 @def_function.function(input_signature=[
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | tracking.py | 266 @def_function.function(input_signature=[], autograph=False)
271 @def_function.function(input_signature=[], autograph=False)
276 @def_function.function(input_signature=[], autograph=False)
|