Searched refs:FusedBatchNormGradOpBase (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 1383 class FusedBatchNormGradOpBase : public OpKernel { class
1385 explicit FusedBatchNormGradOpBase(OpKernelConstruction* context) in FusedBatchNormGradOpBase() function in tensorflow::FusedBatchNormGradOpBase
1505 class FusedBatchNormGradOp : public FusedBatchNormGradOpBase<Device, T, U> {
1508 : FusedBatchNormGradOpBase<Device, T, U>(context) {} in FusedBatchNormGradOp()
1511 FusedBatchNormGradOpBase<Device, T, U>::ComputeWithReservedSpace(context, in Compute()
1517 class FusedBatchNormGradOpV3 : public FusedBatchNormGradOpBase<Device, T, U> {
1520 : FusedBatchNormGradOpBase<Device, T, U>(context) {} in FusedBatchNormGradOpV3()
1523 FusedBatchNormGradOpBase<Device, T, U>::ComputeWithReservedSpace(context, in Compute()
|