HomeSort by relevance Sort by last modified time
    Searched refs:MatrixBandPartFunctor (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
matrix_band_part_op.h 27 struct MatrixBandPartFunctor {
matrix_band_part_op_gpu.cu.cc 52 struct MatrixBandPartFunctor<GPUDevice, Scalar> {
68 #define DEFINE_GPU_SPEC(T) template struct MatrixBandPartFunctor<GPUDevice, T>;
matrix_band_part_op.cc 103 functor::MatrixBandPartFunctor<Device, T> fn;
136 struct MatrixBandPartFunctor<CPUDevice, Scalar> {
192 #define DEFINE_CPU_SPEC(T) template struct MatrixBandPartFunctor<CPUDevice, T>;
204 struct MatrixBandPartFunctor<GPUDevice, T> { \
210 extern template struct MatrixBandPartFunctor<GPUDevice, T>;
cholesky_op.cc 81 struct MatrixBandPartFunctor<GPUDevice, T> { \
87 extern template struct MatrixBandPartFunctor<GPUDevice, T>;
138 functor::MatrixBandPartFunctor<GPUDevice, Scalar> band_part;
qr_op_impl.h 239 functor::MatrixBandPartFunctor<GPUDevice, Scalar> band_part;

Completed in 1590 milliseconds