Searched refs:q_reshaped (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | qr_op_impl.h | 252 auto q_reshaped = q->flat_inner_dims<Scalar, 3>(); in ComputeAsync() local
253 eye(device, q_reshaped); in ComputeAsync()
262 &tau_matrix(batch, 0), &q_reshaped(batch, 0, 0), m, in ComputeAsync()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | dense_attention.py | 496 q_reshaped = array_ops.expand_dims(query, axis=-2)
504 scale * math_ops.tanh(q_reshaped + k_reshaped), axis=-1)
|