Searched defs:FusedBatchNormFreezeGrad (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.h | 66 struct FusedBatchNormFreezeGrad { struct
67 void operator()(OpKernelContext* context, const Tensor& y_backprop_input, in operator()
|
| D | fused_batch_norm_op.cu.cc | 35 struct FusedBatchNormFreezeGrad<GPUDevice, T, U> { struct
36 void operator()(OpKernelContext* context, const Tensor& y_backprop_input, in operator ()()
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
517 void operator()(OpKernelContext* context, const Tensor& y_backprop_input, in operator ()()
|