Searched refs:tau_matrix (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | qr_op_impl.h | 206 auto tau_matrix = tau.matrix<Scalar>(); in ComputeAsync() local
212 &tau_matrix(batch, 0), in ComputeAsync()
262 &tau_matrix(batch, 0), &q_reshaped(batch, 0, 0), m, in ComputeAsync()
280 &tau_matrix(batch, 0), dev_info.back().mutable_data() + batch), in ComputeAsync()
|