Searched refs:bound_dims (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | multi_head_attention.py | 93 def _build_proj_equation(free_dims, bound_dims, output_dims): argument
106 for i in range(bound_dims):
111 letter_offset += bound_dims
349 free_dims, bound_dims=1, output_dims=2)
358 self._key_shape.rank - 1, bound_dims=1, output_dims=2)
367 self._value_shape.rank - 1, bound_dims=1, output_dims=2)
388 free_dims, bound_dims=2, output_dims=len(output_shape))
|