Searched refs:r_reshaped (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | qr_op_impl.h | 207 auto r_reshaped = r->flat_inner_dims<Scalar, 3>(); in ComputeAsync() local
242 &beta, dummy, n, &r_reshaped(batch, 0, 0), n), in ComputeAsync()
252 -1 /* num_upper_diags */, r_reshaped_const, r_reshaped); in ComputeAsync()
|