Searched defs:FusedBatchNormInferenceFunctor (Results 1 – 2 of 2) sorted by relevance
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
49 struct FusedBatchNormInferenceFunctor { struct