Searched defs:TruncatedNormalFunctor (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | parameterized_truncated_normal_op_gpu.cu.cc | 234 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
|
D | parameterized_truncated_normal_op.h | 36 struct TruncatedNormalFunctor { struct
|
D | parameterized_truncated_normal_op.cc | 51 struct TruncatedNormalFunctor<CPUDevice, T> { struct 52 void operator()(OpKernelContext* ctx, const CPUDevice& d, int64 num_batches, in operator ()()
|