Home
last modified time | relevance | path

Searched defs: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
235 void operator()(OpKernelContext* ctx, const GPUDevice& d, int64 num_batches, in operator ()()
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
52 void operator()(OpKernelContext* ctx, const CPUDevice& d, int64 num_batches, in operator ()()