Searched refs:ComputeWithReservedSpace (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 1242 virtual void ComputeWithReservedSpace(OpKernelContext* context, in ComputeWithReservedSpace() function in tensorflow::FusedBatchNormOpBase
1352 FusedBatchNormOpBase<Device, T, U>::ComputeWithReservedSpace(context, in Compute()
1364 FusedBatchNormOpBase<Device, T, U>::ComputeWithReservedSpace(context, true); in Compute()
1378 FusedBatchNormOpBase<Device, T, U>::ComputeWithReservedSpace(context, true); in Compute()
1397 virtual void ComputeWithReservedSpace(OpKernelContext* context, in ComputeWithReservedSpace() function in tensorflow::FusedBatchNormGradOpBase
1511 FusedBatchNormGradOpBase<Device, T, U>::ComputeWithReservedSpace(context, in Compute()
1523 FusedBatchNormGradOpBase<Device, T, U>::ComputeWithReservedSpace(context, in Compute()
|