Searched refs:band_part (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dcholesky_op.cc138 functor::MatrixBandPartFunctor<GPUDevice, Scalar> band_part; variable
139 band_part(context, context->eigen_device<GPUDevice>(),
H A Dqr_op_impl.h236 functor::MatrixBandPartFunctor<GPUDevice, Scalar> band_part; variable
239 band_part(context, device, 0 /* num_lower_diags */,
/external/tensorflow/tensorflow/python/ops/linalg/
H A Dlinalg_impl.py30 band_part = array_ops.matrix_band_part variable

Completed in 993 milliseconds