Home
last modified time | relevance | path

Searched refs:packed_triangular_factors (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dlu_op_gpu.cu.cc124 Tensor* packed_triangular_factors; in ComputeAsync() local
127 {0}, 0, input.shape(), &packed_triangular_factors), in ComputeAsync()
129 if (!packed_triangular_factors->SharesBufferWith(input)) { in ComputeAsync()
130 device.memcpy(packed_triangular_factors->flat<Scalar>().data(), in ComputeAsync()
174 DoMatrixTranspose(device, *packed_triangular_factors, in ComputeAsync()
219 packed_triangular_factors), in ComputeAsync()
Dlu_op.cc139 MatrixMap packed_triangular_factors( in ComputeTensorSlice() local
158 packed_triangular_factors = lu_decomposition.matrixLU(); in ComputeTensorSlice()
170 packed_triangular_factors.diagonal().cwiseAbs().minCoeff(); in ComputeTensorSlice()