| /external/tensorflow/tensorflow/python/keras/engine/ |
| __init__.py | 26 from tensorflow.python.keras.engine.input_spec import InputSpec
|
| input_spec.py | 16 """Contains the InputSpec class."""
28 @keras_export('keras.layers.InputSpec', v1=['keras.layers.InputSpec'])
29 @tf_export(v1=['layers.InputSpec'])
30 class InputSpec(object):
34 a list of instances of InputSpec (one per input tensor).
74 return 'InputSpec(%s)' % ', '.join(x for x in spec if x)
84 input_spec: An InputSpec instance, or None.
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| local.py | 27 from tensorflow.python.keras.engine.input_spec import InputSpec
156 self.input_spec = InputSpec(ndim=3)
220 self.input_spec = InputSpec(ndim=3, axes={1: input_dim})
222 self.input_spec = InputSpec(ndim=3, axes={-1: input_dim})
431 self.input_spec = InputSpec(ndim=4)
503 self.input_spec = InputSpec(ndim=4, axes={1: input_filter})
505 self.input_spec = InputSpec(ndim=4, axes={-1: input_filter})
|
| pooling.py | 26 from tensorflow.python.keras.engine.input_spec import InputSpec
70 self.input_spec = InputSpec(ndim=3)
234 self.input_spec = InputSpec(ndim=4)
421 self.input_spec = InputSpec(ndim=5)
582 self.input_spec = InputSpec(ndim=3)
691 self.input_spec = InputSpec(ndim=4)
782 self.input_spec = InputSpec(ndim=5)
[all...] |
| core.py | 39 from tensorflow.python.keras.engine.input_spec import InputSpec
211 self.input_spec = InputSpec(ndim=3)
268 self.input_spec = InputSpec(ndim=4)
326 self.input_spec = InputSpec(ndim=5)
513 self.input_spec = InputSpec(ndim=len(self.dims) + 1)
567 self.input_spec = InputSpec(min_ndim=1)
631 self.input_spec = InputSpec(ndim=2)
[all...] |
| convolutional_recurrent.py | 30 from tensorflow.python.keras.engine.input_spec import InputSpec
180 self.input_spec = [InputSpec(ndim=5)]
238 self.input_spec[0] = InputSpec(shape=(batch_size, None) + input_shape[2:5])
269 self.state_spec = [InputSpec(shape=(None, dim, None, None))
272 self.state_spec = [InputSpec(shape=(None, None, None, dim))
313 self.state_spec.append(InputSpec(shape=shape))
319 self.constants_spec = [InputSpec(shape=K.int_shape(constant))
[all...] |
| convolutional.py | 30 from tensorflow.python.keras.engine.input_spec import InputSpec
144 self.input_spec = InputSpec(ndim=self.rank + 2)
177 self.input_spec = InputSpec(ndim=self.rank + 2,
[all...] |
| wrappers.py | 27 from tensorflow.python.keras.engine.input_spec import InputSpec
198 self.input_spec = InputSpec(shape=[None, None] + input_shape[2:])
463 state_specs = [InputSpec(shape=K.int_shape(state))
471 constants_spec = [InputSpec(shape=K.int_shape(constant))
|
| advanced_activations.py | 26 from tensorflow.python.keras.engine.input_spec import InputSpec
138 self.input_spec = InputSpec(ndim=len(input_shape), axes=axes)
|
| cudnn_recurrent.py | 28 from tensorflow.python.keras.engine.input_spec import InputSpec
72 self.input_spec = [InputSpec(ndim=3)]
77 self.state_spec = [InputSpec(shape=(None, dim)) for dim in state_size]
|
| kernelized.py | 168 self.input_spec = input_spec.InputSpec(
|
| recurrent.py | 34 from tensorflow.python.keras.engine.input_spec import InputSpec
518 return InputSpec(shape=tuple(input_spec_shape))
570 InputSpec(shape=[None] + tensor_shape.as_shape(dim).as_list())
650 InputSpec(shape=K.int_shape(state)) for state in initial_state
656 InputSpec(shape=K.int_shape(constant)) for constant in constants
[all...] |
| __init__.py | 25 from tensorflow.python.keras.engine.input_spec import InputSpec
|
| recurrent_v2.py | 30 from tensorflow.python.keras.engine.input_spec import InputSpec
576 InputSpec(shape=(None, dim)) for dim in (self.units, self.units)
[all...] |
| normalization.py | 31 from tensorflow.python.keras.engine.input_spec import InputSpec
314 self.input_spec = InputSpec(ndim=ndims, axes=axis_to_dim)
[all...] |
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| core_layers.py | 123 self.input_spec = input_spec.InputSpec(ndim=self.rank + 2)
175 self.input_spec = input_spec.InputSpec(
397 self.input_spec = input_spec.InputSpec(min_ndim=2)
404 self.input_spec = input_spec.InputSpec(
|
| /external/tensorflow/tensorflow/python/layers/ |
| layers.py | 27 from tensorflow.python.keras.engine.input_spec import InputSpec
|
| base_test.py | 271 self.input_spec = input_spec.InputSpec(ndim=2)
298 self.input_spec = input_spec.InputSpec(min_ndim=2)
326 self.input_spec = input_spec.InputSpec(max_ndim=2)
354 self.input_spec = input_spec.InputSpec(dtype='float32')
374 self.input_spec = input_spec.InputSpec(axes={-1: 2})
396 self.input_spec = input_spec.InputSpec(shape=(None, 3))
|
| base.py | 37 InputSpec = base_layer.InputSpec # pylint: disable=invalid-name
185 input_spec: Optional (list of) `InputSpec` object(s) specifying the
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| gru_ops.py | 168 self.input_spec = input_spec.InputSpec(ndim=2)
|
| lstm_ops.py | 389 self.input_spec = input_spec.InputSpec(ndim=2)
632 self.input_spec = input_spec.InputSpec(ndim=3)
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| rnn_cell.py | 77 self.input_spec = base_layer.InputSpec(ndim=2)
255 self.input_spec = base_layer.InputSpec(ndim=2)
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/ |
| cudnn_rnn.py | 326 self.input_spec = input_spec.InputSpec(ndim=3, axes={-1: self._input_size})
|
| /external/tensorflow/tensorflow/python/ops/ |
| rnn_cell_impl.py | 427 self.input_spec = input_spec.InputSpec(ndim=2)
526 self.input_spec = input_spec.InputSpec(ndim=2)
704 self.input_spec = input_spec.InputSpec(ndim=2)
[all...] |
| /external/tensorflow/tensorflow/contrib/keras/api/keras/layers/ |
| __init__.py | 23 from tensorflow.python.keras.engine.input_spec import InputSpec
|
