- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for _FusedBatchNormEx (0.66 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/gpu_fusion.mlir
// Since the tf.AddV2 op has two uses, we have a _FusedBatchNormEx without the // Relu activation and we only fuse the add. // CHECK-NEXT: %[[Y:[a-z0-9]*]], {{.*}}_FusedBatchNormEx // CHECK-NEXT: %[[relu:[a-z0-9]*]] ={{.*}}Relu"(%[[Y]] // CHECK-NEXT: return %[[relu]]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 24 05:47:26 UTC 2022 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/gpu_fusion.cc
// %0 = "tf.Relu"(%y#0) // -> // %y:6 = "tf._FusedBatchNormEx"(%x, %scale, %offset, %mean, %variance) // // Or: // %y:6 = "tf.FusedBatchNormV3"(%x, %scale, %offset, %mean, %variance) // %0 = "tf.AddV2"(%y#0, %side_input) // %1 = "tf.Relu"(%0) // -> // %y:6 = "tf._FusedBatchNormEx"(%x, %scale, %offset, %mean, %variance, // %side_input)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 5.2K bytes - Viewed (0)