- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 82 for Rsqrt (0.05 sec)
-
tensorflow/compiler/mlir/lite/tests/prepare-tf.mlir
// rsqrt(variance + epsilon) // CHECK: %[[RSQRT:.*]] = "tf.Rsqrt"(%[[ADD1]]) // scale * rsqrt(variance + epsilon) // CHECK: %[[MUL1:.*]] = "tf.Mul"(%[[ARG1:.*]], %[[RSQRT]]) // x * scale * rsqrt(variance + epsilon) // CHECK: %[[MUL2:.*]] = "tf.Mul"(%[[ARG0:.*]], %[[MUL1]]) // mean * scale * rsqrt(variance + epsilon)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 07:26:59 UTC 2024 - 59.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/unfuse_mhlo_batch_norm.mlir
// CHECK-DAG: %[[EPS_BCAST:.+]] = mhlo.constant dense<1.001000e-05> : tensor<256xf32> // CHECK-DAG: %[[VARIANCE_EPS:.+]] = mhlo.add %[[VARIANCE]], %[[EPS_BCAST]] : tensor<256xf32> // CHECK-DAG: %[[VARIANCE_EPS_RSQRT:.+]] = mhlo.rsqrt %[[VARIANCE_EPS]] : tensor<256xf32> // CHECK-DAG: %[[MULTIPLIER:.+]] = mhlo.multiply %[[VARIANCE_EPS_RSQRT]], %[[SCALE]] : tensor<256xf32> // CHECK-DAG: %[[MUL_MEAN:.+]] = mhlo.multiply %[[MULTIPLIER]], %[[MEAN]] : tensor<256xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_patterns.td
$t, $m, $v, $beta, $gamma, F32Attr:$variance_epsilon, ConstBoolAttrFalse:$scale_after_normalization), (TF_AddOp (TF_MulOp $t, (TF_RsqrtOp:$rsqrt (TF_AddOp $v, (TF_ConstOp $variance_epsilon)))), (TF_SubOp $beta, (TF_MulOp $m, $rsqrt)))>; def ConvertBatchNormWithGlobalNormalization_2 : Pat< (TF_BatchNormWithGlobalNormalizationOp
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/vhlo.mlir
//CHECK-NEXT: return %0 : tensor<1x30x1xi32> //CHECK-NEXT:} func.func @rsqrt(%arg0: tensor<1x1x1x96xf32>) -> tensor<1x1x1x96xf32> { %0 = "vhlo.rsqrt_v1" (%arg0) : (tensor<1x1x1x96xf32>) -> tensor<1x1x1x96xf32> func.return %0 : tensor<1x1x1x96xf32> } //CHECK:func.func private @rsqrt(%arg0: tensor<1x1x1x96xf32>) -> tensor<1x1x1x96xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 14 19:15:40 UTC 2024 - 31.9K bytes - Viewed (1) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.td
$round_val ), /*incompatible_shape_error*/ConstBoolAttrTrue ), $zero, $rounded )>; //===----------------------------------------------------------------------===// // Rsqrt op patterns. //===----------------------------------------------------------------------===// // RsqrtGrad(lhs, rhs) = (lhs * lhs * lhs) * (rhs / -2) def LowerRsqrtGradOp : Pat< (TF_RsqrtGradOp $lhs, $rhs),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 13:30:42 UTC 2024 - 24.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/const-fold.mlir
%8 = "tfl.sin"(%1) : (tensor<f32>) -> tensor<f32> %9 = "tfl.cos"(%2) : (tensor<f32>) -> tensor<f32> %10 = "tfl.log"(%3) : (tensor<f32>) -> tensor<f32> %11 = "tfl.sqrt"(%4) : (tensor<f32>) -> tensor<f32> %12 = "tfl.rsqrt"(%5) : (tensor<f32>) -> tensor<f32> %13 = "tfl.square"(%6) : (tensor<f32>) -> tensor<f32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 45.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td
// constraint on axis and depth. multiclass L2NormalizePatterns<Op FirstOp, Op SecondOp> { // This pattern constructs L2NormalizationOp from // Mul->Rsqrt->Sum->Square Or // Div->sqrt->Sum->Square def L2NormalizePattern1#FirstOp#SecondOp : Pat< (FirstOp $x, (SecondOp (TFL_SumOp
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 66.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// CHECK: mhlo.rsqrt %arg0 : tensor<?xf32> %0 = "tf.Rsqrt"(%arg0) : (tensor<?xf32>) -> tensor<?xf32> func.return %0 : tensor<?xf32> } // ----- // CHECK-LABEL: func @sqrt func.func @sqrt(%arg0: tensor<2xf32>) -> tensor<2xf32> { // CHECK: mhlo.sqrt %arg0 : tensor<2xf32> %0 = "tf.Sqrt"(%arg0) : (tensor<2xf32>) -> tensor<2xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/schema_v3b.fbs
SELECT = 64, SLICE = 65, SIN = 66, TRANSPOSE_CONV = 67, SPARSE_TO_DENSE = 68, TILE = 69, EXPAND_DIMS = 70, EQUAL = 71, NOT_EQUAL = 72, LOG = 73, SUM = 74, SQRT = 75, RSQRT = 76, SHAPE = 77, POW = 78, ARG_MIN = 79, FAKE_QUANT = 80, REDUCE_PROD = 81, REDUCE_MAX = 82, PACK = 83, LOGICAL_OR = 84, ONE_HOT = 85, LOGICAL_AND = 86,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 14:28:27 UTC 2024 - 30K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
%0 = "tf.Sqrt"(%arg1) : (tensor<8x16xf32>) -> tensor<8x16xf32> %1 = "tf.Div"(%arg0, %0) {device = "/job:localhost/replica:0/task:0/device:GPU:0"} : (tensor<8x16xf32>, tensor<8x16xf32>) -> tensor<8x16xf32> func.return %1: tensor<8x16xf32> // CHECK: %0 = "tf.Rsqrt"(%arg1) {device = "/job:localhost/replica:0/task:0/device:GPU:0"} : (tensor<8x16xf32>) -> tensor<8x16xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0)