- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for _FusedBatchNormEx (0.24 sec)
-
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) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
{"REDUCEWINDOWPW", {"BiasAddGrad", "LRN", "LRNGrad"}}, {"BN", {"FusedBatchNorm", "FusedBatchNormV2", "FusedBatchNormV3", "_FusedBatchNormEx", "FusedBatchNormGrad", "FusedBatchNormGradV2", "FusedBatchNormGradV3"}}, {"Conv", {"_FusedConv2D"}}, {"SORT", {"TopKV2"}}, // XLA version much faster then TF version.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
); let results = (outs TF_Tensor:$output ); TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>; } def TF__FusedBatchNormExOp : TF_Op<"_FusedBatchNormEx", [Pure]> { let summary = "Internal FusedBatchNorm operation: reserved for internal use."; let description = [{ Do not invoke this operator directly in Python. A fusion optimization is
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)