Searched refs:shared_axes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | advanced_activations.py | 106 shared_axes=None, argument
113 if shared_axes is None:
114 self.shared_axes = None
115 elif not isinstance(shared_axes, (list, tuple)):
116 self.shared_axes = [shared_axes]
118 self.shared_axes = list(shared_axes)
123 if self.shared_axes is not None:
124 for i in self.shared_axes:
134 if self.shared_axes:
136 if i not in self.shared_axes:
[all …]
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | ops.py | 184 shared_axes = axes_0.remove(axis_name)
190 if current_shared_axes != shared_axes:
195 (shared_axes, current_shared_axes))
815 shared_axes = set(a.axes) & set(b.axes)
816 if len(shared_axes) > 1:
820 'over.' % shared_axes)
821 if not shared_axes:
825 shared_axis, = shared_axes
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.layers.-p-re-l-u.pbtxt | 89 … \'alpha_initializer\', \'alpha_regularizer\', \'alpha_constraint\', \'shared_axes\'], varargs=Non…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.layers.-p-re-l-u.pbtxt | 89 … \'alpha_initializer\', \'alpha_regularizer\', \'alpha_constraint\', \'shared_axes\'], varargs=Non…
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples.py | 809 prelu = tf.keras.layers.PReLU(shared_axes=parameters["shared_axes"])
819 shared_axes = parameters["shared_axes"]
823 alpha_shape.append(1 if dim in shared_axes else input_shape[dim])
|