Searched refs:FusedBatchNormOpV3 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 205 class FusedBatchNormOpV3 : public FusedBatchNormOp { class
207 explicit FusedBatchNormOpV3(OpKernelConstruction* ctx) in FusedBatchNormOpV3() function in tensorflow::__anonfc91c4b10111::FusedBatchNormOpV3
235 REGISTER_XLA_OP(Name("FusedBatchNormV3"), FusedBatchNormOpV3);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 1358 class FusedBatchNormOpV3 : public FusedBatchNormOpBase<Device, T, U> { class
1360 explicit FusedBatchNormOpV3(OpKernelConstruction* context) in FusedBatchNormOpV3() function in tensorflow::FusedBatchNormOpV3
1564 FusedBatchNormOpV3<CPUDevice, float, float>);
1576 FusedBatchNormOpV3<CPUDevice, Eigen::half, float>);
1622 FusedBatchNormOpV3<GPUDevice, float, float>);
1640 FusedBatchNormOpV3<GPUDevice, Eigen::half, float>);
|