- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for RSQRT (0.04 sec)
-
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
%0 = "tf.Sin"(%arg0) : (tensor<?xf32>) -> tensor<?xf32> func.return %0 : tensor<?xf32> } // ----- // CHECK-LABEL: func @rsqrt func.func @rsqrt(%arg0: tensor<2xf32>) -> tensor<2xf32> { // CHECK: mhlo.rsqrt %arg0 : tensor<2xf32> %0 = "tf.Rsqrt"(%arg0) : (tensor<2xf32>) -> tensor<2xf32> func.return %0 : tensor<2xf32> } // ----- // CHECK-LABEL: func @rsqrt_dynamic
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/stablehlo/tests/legalize_hlo.mlir
} // CHECK-LABEL: func @rsqrt( // CHECK-SAME: %[[VAL_0:.*]]: tensor<2xf32>) -> tensor<2xf32> { // CHECK: %[[VAL_1:.*]] = "tf.Rsqrt"(%[[VAL_0]]) : (tensor<2xf32>) -> tensor<2xf32> // CHECK: return %[[VAL_1]] : tensor<2xf32> // CHECK: } func.func @rsqrt(%arg0: tensor<2xf32>) -> tensor<2xf32> { %0 = "mhlo.rsqrt"(%arg0) : (tensor<2xf32>) -> tensor<2xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 07:26:59 UTC 2024 - 340.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/optimize.mlir
%cst = arith.constant dense<[0]> : tensor<1xi32> %0 = "tfl.square"(%arg0) : (tensor<2xf32>) -> tensor<2xf32> %1 = "tfl.sum"(%0, %cst) {keep_dims = false} : (tensor<2xf32>, tensor<1xi32>) -> tensor<f32> %2 = "tfl.rsqrt"(%1) : (tensor<f32>) -> tensor<f32> %3 = "tfl.mul"(%arg0, %2) {fused_activation_function = "NONE"} : (tensor<2xf32>, tensor<f32>) -> tensor<2xf32> func.return %3: tensor<2xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 284.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
non_feature_dims.push_back(i); } auto reduce_dims = GetI64ElementsAttr(non_feature_dims, &rewriter); auto scalar_broadcast_dims = rewriter.getDenseI64ArrayAttr({}); // scratch1 = rsqrt(var + epsilon) RankedTensorType scalar_float = tensorflow::GetTypeFromTFTensorShape({}, kernel_type); auto epsilon = rewriter.create<ConstantOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)