Searched defs:FusedBatchNormInferenceFunctor (Results 1 – 2 of 2) sorted by relevance
49 struct FusedBatchNormInferenceFunctor { struct
318 struct FusedBatchNormInferenceFunctor<GPUDevice, T, U> { struct319 void operator()(OpKernelContext* context, TensorFormat tensor_format, in operator ()()399 template struct FusedBatchNormInferenceFunctor<GPUDevice, float, float>; variable400 template struct FusedBatchNormInferenceFunctor<GPUDevice, Eigen::half, float>; variable