Searched refs:FusedBatchNormOpEx (Results 1 – 2 of 2) sorted by relevance
219 class FusedBatchNormOpEx : public FusedBatchNormOp { class221 explicit FusedBatchNormOpEx(OpKernelConstruction* ctx) in FusedBatchNormOpEx() function in tensorflow::__anonfc91c4b10111::FusedBatchNormOpEx236 REGISTER_XLA_OP(Name("_FusedBatchNormEx"), FusedBatchNormOpEx);
1369 class FusedBatchNormOpEx : public FusedBatchNormOpBase<Device, T, U> { class1373 explicit FusedBatchNormOpEx(OpKernelConstruction* context) in FusedBatchNormOpEx() function in tensorflow::FusedBatchNormOpEx1628 FusedBatchNormOpEx<GPUDevice, float, float>);1646 FusedBatchNormOpEx<GPUDevice, Eigen::half, float>);