Home
last modified time | relevance | path

Searched defs:MatrixBandPartFunctor (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_band_part_op_gpu.cu.cc52 struct MatrixBandPartFunctor<GPUDevice, Scalar> { struct
53 void operator()(OpKernelContext* context, const GPUDevice& device, in operator ()()
Dmatrix_band_part_op.h27 struct MatrixBandPartFunctor { struct
Dmatrix_band_part_op.cc137 struct MatrixBandPartFunctor<CPUDevice, Scalar> { struct
138 void operator()(OpKernelContext* context, const CPUDevice& device, in operator ()()