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 ()()
|