Searched refs:_output_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | multi_head_attention.py | 239 self._output_shape = output_shape
267 self._output_shape,
380 if self._output_shape:
381 if not isinstance(self._output_shape, collections.abc.Sized):
382 output_shape = [self._output_shape]
384 output_shape = self._output_shape
|
| D | core.py | 864 self._output_shape = output_shape
875 if self._output_shape is None:
888 if callable(self._output_shape):
889 output_shapes = self._output_shape(input_shape)
899 output_shapes = tf_utils.convert_shapes(self._output_shape, to_tuples=False)
976 output_shape_config = self._serialize_function_to_config(self._output_shape,
|
| D | core_test.py | 223 self.assertAllEqual(layer._output_shape, (1, 1))
228 self.assertAllEqual(layer._output_shape, (1, 1))
|
| /external/tensorflow/tensorflow/python/keras/feature_column/ |
| D | base_feature_layer.py | 83 def _output_shape(self, input_shape, num_elements): member in _BaseFeaturesLayer
|