Searched refs:FusedBatchNormFreezeGrad (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.h | 66 struct FusedBatchNormFreezeGrad { struct
|
| D | fused_batch_norm_op.cu.cc | 35 struct FusedBatchNormFreezeGrad<GPUDevice, T, U> { struct
108 template struct FusedBatchNormFreezeGrad<GPUDevice, float, float>; variable
109 template struct FusedBatchNormFreezeGrad<GPUDevice, Eigen::half, float>; variable
|
| D | fused_batch_norm_op.cc | 516 struct FusedBatchNormFreezeGrad<CPUDevice, T, U> { struct
1152 void FusedBatchNormFreezeGrad<GPUDevice, T, U>::operator()( \
1158 extern template struct FusedBatchNormFreezeGrad<GPUDevice, T, U>; \
1487 functor::FusedBatchNormFreezeGrad<Device, T, U>()( in ComputeWithReservedSpace()
|