Searched defs:FusedBatchNormGrad (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 349 struct FusedBatchNormGrad<CPUDevice, T, U> { struct
350 void operator()(OpKernelContext* context, const Tensor& y_backprop_input, in operator ()()
994 struct FusedBatchNormGrad<GPUDevice, T, U> { struct
995 void operator()(OpKernelContext* context, const Tensor& y_backprop, in operator ()()
|
| D | fused_batch_norm_op_test.cc | 261 static Graph* FusedBatchNormGrad(int n, int h, int w, int c, bool is_training, in FusedBatchNormGrad() function
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 17790 func FusedBatchNormGrad(scope *Scope, y_backprop tf.Output, x tf.Output, scale tf.Output, reserve_s… func
|