Searched refs:MatrixBandPartFunctor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_band_part_op.cc | 104 functor::MatrixBandPartFunctor<Device, T> fn; in Compute()
137 struct MatrixBandPartFunctor<CPUDevice, Scalar> { struct
194 #define DEFINE_CPU_SPEC(T) template struct MatrixBandPartFunctor<CPUDevice, T>;
206 struct MatrixBandPartFunctor<GPUDevice, T> { \
212 extern template struct MatrixBandPartFunctor<GPUDevice, T>;
|
| D | cholesky_op.cc | 81 struct MatrixBandPartFunctor<GPUDevice, T> { \
87 extern template struct MatrixBandPartFunctor<GPUDevice, T>;
137 functor::MatrixBandPartFunctor<GPUDevice, Scalar> band_part; in ComputeAsync()
177 functor::MatrixBandPartFunctor<GPUDevice, Scalar> band_part; in ComputeAsync()
|
| D | matrix_band_part_op_gpu.cu.cc | 52 struct MatrixBandPartFunctor<GPUDevice, Scalar> { struct
69 #define DEFINE_GPU_SPEC(T) template struct MatrixBandPartFunctor<GPUDevice, T>;
|
| D | matrix_band_part_op.h | 27 struct MatrixBandPartFunctor { struct
|
| D | qr_op_impl.h | 239 functor::MatrixBandPartFunctor<GPUDevice, Scalar> band_part; in ComputeAsync()
|