Searched defs:FusedBatchNormFreezeGrad (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.h | 60 struct FusedBatchNormFreezeGrad { struct
61 void operator()(OpKernelContext* context, const Tensor& y_backprop_input, in operator()
|
| D | fused_batch_norm_op.cu.cc | 36 struct FusedBatchNormFreezeGrad<GPUDevice, T, U> { struct
37 void operator()(OpKernelContext* context, const Tensor& y_backprop_input, in operator ()()
115 template struct FusedBatchNormFreezeGrad<GPUDevice, float, float>; variable
116 template struct FusedBatchNormFreezeGrad<GPUDevice, Eigen::half, float>; variable
|
| D | fused_batch_norm_op.cc | 531 struct FusedBatchNormFreezeGrad<CPUDevice, T, U> { struct
532 void operator()(OpKernelContext* context, const Tensor& y_backprop_input, in operator ()()
|