HomeSort by relevance Sort by last modified time
    Searched defs:MatrixBandPartFunctor (Results 1 - 3 of 3) 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>;

Completed in 145 milliseconds