Searched refs:FusedBatchNormGradOpV3 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 1517 class FusedBatchNormGradOpV3 : public FusedBatchNormGradOpBase<Device, T, U> { class
1519 explicit FusedBatchNormGradOpV3(OpKernelConstruction* context) in FusedBatchNormGradOpV3() function in tensorflow::FusedBatchNormGradOpV3
1570 FusedBatchNormGradOpV3<CPUDevice, float, float>);
1582 FusedBatchNormGradOpV3<CPUDevice, Eigen::half, float>);
1634 FusedBatchNormGradOpV3<GPUDevice, float, float>);
1652 FusedBatchNormGradOpV3<GPUDevice, Eigen::half, float>);
|