Searched refs:MklFusedBatchNormOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 639 class MklFusedBatchNormOp : public OpKernel { class
641 explicit MklFusedBatchNormOp(OpKernelConstruction* context) in MklFusedBatchNormOp() function in tensorflow::MklFusedBatchNormOp
1394 MklFusedBatchNormOp<CPUDevice, T, T, false, false>); \
1400 MklFusedBatchNormOp<CPUDevice, T, T, false, false, true>);
1413 MklFusedBatchNormOp<CPUDevice, T, U, false, false>); \
1420 MklFusedBatchNormOp<CPUDevice, T, U, false, false, true>);
1474 MklFusedBatchNormOp<CPUDevice, T, U, true, false>); \
1481 MklFusedBatchNormOp<CPUDevice, T, U, true, true>); \
1488 MklFusedBatchNormOp<CPUDevice, T, U, true, false, true>); \
1495 MklFusedBatchNormOp<CPUDevice, T, U, true, true, true>);
|