Searched refs:MklFusedBatchNormGradOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 1046 class MklFusedBatchNormGradOp : public OpKernel { class
1048 explicit MklFusedBatchNormGradOp(OpKernelConstruction* context) in MklFusedBatchNormGradOp() function in tensorflow::MklFusedBatchNormGradOp
1432 MklFusedBatchNormGradOp<CPUDevice, T, T, false>); \
1438 MklFusedBatchNormGradOp<CPUDevice, T, T, false, true>);
1451 MklFusedBatchNormGradOp<CPUDevice, T, U, false>); \
1458 MklFusedBatchNormGradOp<CPUDevice, T, U, false, true>);
1519 MklFusedBatchNormGradOp<CPUDevice, T, U, true>); \
1526 MklFusedBatchNormGradOp<CPUDevice, T, U, true, true>);
|