Searched refs:_output_shape (Results 1 – 5 of 5) sorted by relevance
735 self._output_shape = output_shape743 if self._output_shape is None:756 if callable(self._output_shape):757 output_shapes = self._output_shape(input_shape)767 output_shapes = tf_utils.convert_shapes(self._output_shape, to_tuples=False)805 if isinstance(self._output_shape, python_types.LambdaType):806 output_shape = generic_utils.func_dump(self._output_shape)808 output_shape_module = self._output_shape.__module__809 elif callable(self._output_shape):810 output_shape = self._output_shape.__name__[all …]
71 self._output_shape = tuple(kwargs.pop('output_shape'))104 result.set_shape((x.shape[0],) + self._output_shape)
78 self._output_shape = outputs_t.shape147 self._output_shape,168 return self._output_shape
131 self._output_shape = [-1, hidden_dim]146 return self.linear(tf.reshape(y, self._output_shape))
347 def _output_shape(self, input_shape, num_elements): member in _BaseFeaturesLayer