Searched defs:FusedBatchNormInferenceFunctor (Results 1 – 2 of 2) sorted by relevance
316 struct FusedBatchNormInferenceFunctor<GPUDevice, T, U> { struct317 void operator()(OpKernelContext* context, TensorFormat tensor_format, in operator ()()397 template struct FusedBatchNormInferenceFunctor<GPUDevice, float, float>; variable398 template struct FusedBatchNormInferenceFunctor<GPUDevice, Eigen::half, float>; variable
43 struct FusedBatchNormInferenceFunctor { struct