Home
last modified time | relevance | path

Searched refs:input_spec (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/python/keras/engine/
Dinput_spec.py77 def assert_input_compatibility(input_spec, inputs, layer_name): argument
92 if not input_spec:
94 if not isinstance(input_spec, (list, tuple)):
95 input_spec = nest.flatten(input_spec)
98 if len(inputs) != len(input_spec):
100 str(len(input_spec)) + ' inputs, '
103 for input_index, (x, spec) in enumerate(zip(inputs, input_spec)):
D__init__.py26 from tensorflow.python.keras.engine.input_spec import InputSpec
Dbase_layer.py41 from tensorflow.python.keras.engine import input_spec
149 self.input_spec = None
582 input_spec.assert_input_compatibility(self.input_spec, inputs,
1696 input_spec.assert_input_compatibility(
1697 self.input_spec, inputs, self.name)
2067 InputSpec = input_spec.InputSpec # pylint:disable=invalid-name
Dsequential.py361 def input_spec(self): member in Sequential
363 return self.layers[0].input_spec
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Dcore_layers.py24 from tensorflow.python.keras.engine import input_spec
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/
Dbase_test.py30 from tensorflow.python.keras.engine import input_spec
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))
416 self.input_spec = None
Dlayers.py27 from tensorflow.python.keras.engine.input_spec import InputSpec
/external/tensorflow/tensorflow/python/keras/layers/
Dpooling.py26 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)
Dlocal.py27 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})
Dwrappers.py27 from tensorflow.python.keras.engine.input_spec import InputSpec
198 self.input_spec = InputSpec(shape=[None, None] + input_shape[2:])
412 self.input_spec = layer.input_spec
499 original_input_spec = self.input_spec
500 self.input_spec = full_input_spec
502 self.input_spec = original_input_spec
Dcore.py39 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)
977 self.input_spec = InputSpec(min_ndim=2)
989 self.input_spec = InputSpec(min_ndim=2,
Dkernelized.py29 from tensorflow.python.keras.engine import input_spec
168 self.input_spec = input_spec.InputSpec(
Dconvolutional.py30 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,
754 self.input_spec = InputSpec(ndim=4, axes={channel_axis: input_dim})
1028 self.input_spec = InputSpec(ndim=5, axes={channel_axis: input_dim})
1287 self.input_spec = InputSpec(ndim=self.rank + 2,
1804 self.input_spec = InputSpec(ndim=4, axes={channel_axis: input_dim})
1884 self.input_spec = InputSpec(ndim=3)
1950 self.input_spec = InputSpec(ndim=4)
2019 self.input_spec = InputSpec(ndim=5)
[all …]
Dconvolutional_recurrent.py30 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])
333 full_input_spec = self.input_spec + additional_specs
335 original_input_spec = self.input_spec
336 self.input_spec = full_input_spec
338 self.input_spec = original_input_spec
417 input_shape = self.input_spec[0].shape
Drecurrent.py34 from tensorflow.python.keras.engine.input_spec import InputSpec
406 self.input_spec = None
537 if self.input_spec is not None:
538 self.input_spec[0] = get_input_spec(input_shape)
540 self.input_spec = [get_input_spec(input_shape)]
546 if self.input_spec is not None:
547 self.input_spec[:self._num_inputs] = flat_input_shapes
549 self.input_spec = flat_input_shapes
678 self.input_spec = full_input_spec
683 self.input_spec = self.input_spec[:-len(additional_specs)]
[all …]
Dadvanced_activations.py26 from tensorflow.python.keras.engine.input_spec import InputSpec
138 self.input_spec = InputSpec(ndim=len(input_shape), axes=axes)
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
Dgru_ops.py24 from tensorflow.python.keras.engine import input_spec
168 self.input_spec = input_spec.InputSpec(ndim=2)
Dlstm_ops.py28 from tensorflow.python.keras.engine import input_spec
389 self.input_spec = input_spec.InputSpec(ndim=2)
632 self.input_spec = input_spec.InputSpec(ndim=3)
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dinput_data.h39 TensorDataSet(const tensorforest::TensorForestDataSpec& input_spec, in TensorDataSet() argument
44 input_spec_(input_spec), in TensorDataSet()
/external/tensorflow/tensorflow/python/ops/
Drnn_cell_impl.py40 from tensorflow.python.keras.engine import input_spec
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)
895 self.input_spec = input_spec.InputSpec(ndim=2)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.layers.-input-spec.pbtxt3 is_instance: "<class \'tensorflow.python.keras.engine.input_spec.InputSpec\'>"
Dtensorflow.keras.layers.-input-spec.pbtxt3 is_instance: "<class \'tensorflow.python.keras.engine.input_spec.InputSpec\'>"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.layers.-input-spec.pbtxt3 is_instance: "<class \'tensorflow.python.keras.engine.input_spec.InputSpec\'>"
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/
Dcudnn_rnn.py24 from tensorflow.python.keras.engine import input_spec
326 self.input_spec = input_spec.InputSpec(ndim=3, axes={-1: self._input_size})
/external/tensorflow/tensorflow/contrib/tensor_forest/python/
Dtensor_forest.py630 input_spec=data_spec.SerializeToString(),
650 input_spec=data_spec.SerializeToString(),
689 input_spec=data_spec.SerializeToString(),

12