Home
last modified time | relevance | path

Searched refs:SetNanFunctor (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.h42 struct SetNanFunctor { struct
Dfused_batch_norm_op.cu.cc112 void SetNanFunctor<T>::operator()(const Eigen::GpuDevice& d, in operator ()()
118 template class SetNanFunctor<float>; variable
Dfused_batch_norm_op.cc821 functor::SetNanFunctor<U> f; in operator ()()