Searched refs:free_dims (Results 1 – 2 of 2) 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
100 for i in range(free_dims):
105 letter_offset += free_dims
347 free_dims = self._query_shape.rank - 1
349 free_dims, bound_dims=1, output_dims=2)
388 free_dims, bound_dims=2, output_dims=len(output_shape))
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops.py | 4678 free_dims = [shape_a[i] for i in free]
4691 return reshaped_a, free_dims, free_dims
4698 free_dims = [shape_a[i] for i in free]
4699 free_dims_static = free_dims
4710 free_dims = array_ops.gather(shape_a, free)
4712 prod_free_dims = reduce_prod(free_dims)
4721 return reshaped_a, free_dims, free_dims_static
|