Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dparameterized_truncated_normal_op_gpu.cu.cc234 struct TruncatedNormalFunctor<GPUDevice, T> { struct
255 template struct TruncatedNormalFunctor<GPUDevice, Eigen::half>; variable
256 template struct TruncatedNormalFunctor<GPUDevice, float>; variable
257 template struct TruncatedNormalFunctor<GPUDevice, double>; variable
Dparameterized_truncated_normal_op.h36 struct TruncatedNormalFunctor { struct
Dparameterized_truncated_normal_op.cc51 struct TruncatedNormalFunctor<CPUDevice, T> { struct
422 auto truncFunctor = functor::TruncatedNormalFunctor<Device, T>(); in Compute()