Searched defs:BatchNormOp (Results 1 - 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ | ||
| H A D | batch_norm_op.cc | 36 class BatchNormOp : public OpKernel { class in namespace:tensorflow 38 explicit BatchNormOp(OpKernelConstruction* context) : OpKernel(context) { function in class:tensorflow::BatchNormOp 167 BatchNormOp<CPUDevice, T>); 199 BatchNormOp<GPUDevice, T>); 212 BatchNormOp<SYCLDevice, T>); |
Completed in 99 milliseconds