Home
last modified time | relevance | path

Searched refs:band_part (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcholesky_op.cc138 functor::MatrixBandPartFunctor<GPUDevice, Scalar> band_part; in ComputeAsync() local
139 band_part(context, context->eigen_device<GPUDevice>(), in ComputeAsync()
Dqr_op_impl.h239 functor::MatrixBandPartFunctor<GPUDevice, Scalar> band_part; in ComputeAsync() local
242 band_part(context, device, 0 /* num_lower_diags */, in ComputeAsync()
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_MatrixBandPart.pbtxt4 name: "linalg.band_part"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.linalg.pbtxt72 name: "band_part"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.linalg.pbtxt72 name: "band_part"
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinalg_impl.py34 band_part = array_ops.matrix_band_part variable