Searched refs:FusedBatchNormOpBase (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 1180 class FusedBatchNormOpBase : public OpKernel { class
1184 explicit FusedBatchNormOpBase(OpKernelConstruction* context, in FusedBatchNormOpBase() function in tensorflow::FusedBatchNormOpBase
1346 class FusedBatchNormOp : public FusedBatchNormOpBase<Device, T, U> {
1349 : FusedBatchNormOpBase<Device, T, U>(context) {} in FusedBatchNormOp()
1352 FusedBatchNormOpBase<Device, T, U>::ComputeWithReservedSpace(context, in Compute()
1358 class FusedBatchNormOpV3 : public FusedBatchNormOpBase<Device, T, U> {
1361 : FusedBatchNormOpBase<Device, T, U>(context) {} in FusedBatchNormOpV3()
1364 FusedBatchNormOpBase<Device, T, U>::ComputeWithReservedSpace(context, true); in Compute()
1369 class FusedBatchNormOpEx : public FusedBatchNormOpBase<Device, T, U> {
1374 : FusedBatchNormOpBase<Device, T, U>(context, in FusedBatchNormOpEx()
[all …]
|