Home
last modified time | relevance | path

Searched refs:FusedBatchNormOp (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc31 class FusedBatchNormOp : public XlaOpKernel { class
33 explicit FusedBatchNormOp(OpKernelConstruction* ctx) in FusedBatchNormOp() function in tensorflow::__anonfc91c4b10111::FusedBatchNormOp
34 : FusedBatchNormOp(ctx, false) {} in FusedBatchNormOp()
36 FusedBatchNormOp(OpKernelConstruction* ctx, bool is_batch_norm_ex) in FusedBatchNormOp() function in tensorflow::__anonfc91c4b10111::FusedBatchNormOp
205 class FusedBatchNormOpV3 : public FusedBatchNormOp {
208 : FusedBatchNormOp(ctx) {} in FusedBatchNormOpV3()
211 FusedBatchNormOp::CompileImpl(ctx); in Compile()
219 class FusedBatchNormOpEx : public FusedBatchNormOp {
222 : FusedBatchNormOp(ctx, /*is_batch_norm_ex=*/true) {} in FusedBatchNormOpEx()
225 FusedBatchNormOp::CompileImpl(ctx); in Compile()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cc1346 class FusedBatchNormOp : public FusedBatchNormOpBase<Device, T, U> { class
1348 explicit FusedBatchNormOp(OpKernelConstruction* context) in FusedBatchNormOp() function in tensorflow::FusedBatchNormOp
1530 FusedBatchNormOp<CPUDevice, float, float>);
1540 FusedBatchNormOp<CPUDevice, float, float>);
1552 FusedBatchNormOp<CPUDevice, Eigen::half, float>);
1588 FusedBatchNormOp<GPUDevice, float, float>);
1598 FusedBatchNormOp<GPUDevice, float, float>);
1610 FusedBatchNormOp<GPUDevice, Eigen::half, float>);
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc2587 class ConvertFusedBatchNorm : public OpRewritePattern<TF::FusedBatchNormOp> {
2588 using OpRewritePattern<FusedBatchNormOp>::OpRewritePattern;
2589 LogicalResult matchAndRewrite(TF::FusedBatchNormOp tf_fused_batch_norm_op, in matchAndRewrite()
2611 void FusedBatchNormOp::getCanonicalizationPatterns( in getCanonicalizationPatterns()
Dtf_ops_a_m.cc2329 static LogicalResult Verify(FusedBatchNormOp op) { in Verify()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc1256 auto tf_batchnorm_op = cast<TF::FusedBatchNormOp>(op); in matchAndRewrite()