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

  /external/tensorflow/tensorflow/core/kernels/
fill_functor.h 36 struct SetZeroFunctor {
41 // Partial specialization of SetZeroFunctor<Device=Eigen::ThreadPoolDevice, T>.
43 struct SetZeroFunctor<Eigen::ThreadPoolDevice, T> {
49 // Partial specialization of SetZeroFunctor<Device=Eigen::SyclDevice, T>.
51 struct SetZeroFunctor<Eigen::SyclDevice, T> {
57 struct SetZeroFunctor<Eigen::ThreadPoolDevice, string> {
constant_op_gpu.cu.cc 85 struct SetZeroFunctor<GPUDevice, T> {
91 #define DEFINE_SETZERO_GPU(T) template struct SetZeroFunctor<GPUDevice, T>;
fill_functor.cu.cc 85 struct SetZeroFunctor<GPUDevice, T> {
92 void SetZeroFunctor<GPUDevice, Variant>::operator()(
97 #define DEFINE_SETZERO_GPU(T) template struct SetZeroFunctor<GPUDevice, T>;

Completed in 78 milliseconds