Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/linalg/
Dcholesky_op.cc137 functor::MatrixBandPartFunctor<GPUDevice, Scalar> band_part; in ComputeAsync() local
138 band_part(context, context->eigen_device<GPUDevice>(), in ComputeAsync()
177 functor::MatrixBandPartFunctor<GPUDevice, Scalar> band_part; in ComputeAsync() local
178 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/python/ops/linalg/
Dlinalg_impl.py39 band_part = array_ops.matrix_band_part variable
1004 band_part(lower_upper, num_lower=-1, num_upper=0),
1127 band_part(lower_upper, num_lower=-1, num_upper=0),
1129 upper = band_part(lower_upper, num_lower=0, num_upper=-1)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.linalg.pbtxt96 name: "band_part"