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

  /external/tensorflow/tensorflow/core/kernels/
diag_op.h 28 struct DiagFunctor {
diag_op_gpu.cu.cc 44 struct DiagFunctor<GPUDevice, T> {
77 template struct DiagFunctor<GPUDevice, double>;
78 template struct DiagFunctor<GPUDevice, float>;
79 template struct DiagFunctor<GPUDevice, int32>;
80 template struct DiagFunctor<GPUDevice, int64>;
81 template struct DiagFunctor<GPUDevice, complex64>;
82 template struct DiagFunctor<GPUDevice, complex128>;
diag_op.cc 63 functor::DiagFunctor<Device, T> diagFunc;
128 struct DiagFunctor<CPUDevice, T> {
201 extern template struct DiagFunctor<GPUDevice, double>;
202 extern template struct DiagFunctor<GPUDevice, float>;
203 extern template struct DiagFunctor<GPUDevice, int32>;
204 extern template struct DiagFunctor<GPUDevice, int64>;
205 extern template struct DiagFunctor<GPUDevice, complex64>;
206 extern template struct DiagFunctor<GPUDevice, complex128>;

Completed in 175 milliseconds