Home
last modified time | relevance | path

Searched refs:FusedBatchNormGradOp (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc238 class FusedBatchNormGradOp : public XlaOpKernel { class
240 explicit FusedBatchNormGradOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in FusedBatchNormGradOp() function in tensorflow::__anonfc91c4b10111::FusedBatchNormGradOp
350 REGISTER_XLA_OP(Name("FusedBatchNormGrad"), FusedBatchNormGradOp);
351 REGISTER_XLA_OP(Name("FusedBatchNormGradV2"), FusedBatchNormGradOp);
352 REGISTER_XLA_OP(Name("FusedBatchNormGradV3"), FusedBatchNormGradOp);
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cc1505 class FusedBatchNormGradOp : public FusedBatchNormGradOpBase<Device, T, U> { class
1507 explicit FusedBatchNormGradOp(OpKernelConstruction* context) in FusedBatchNormGradOp() function in tensorflow::FusedBatchNormGradOp
1534 FusedBatchNormGradOp<CPUDevice, float, float>);
1546 FusedBatchNormGradOp<CPUDevice, float, float>);
1558 FusedBatchNormGradOp<CPUDevice, Eigen::half, float>);
1592 FusedBatchNormGradOp<GPUDevice, float, float>);
1604 FusedBatchNormGradOp<GPUDevice, float, float>);
1616 FusedBatchNormGradOp<GPUDevice, Eigen::half, float>);
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1923 ConvertFusedBatchNormGradBase<TF::FusedBatchNormGradOp>;