- Sort Score
- Result 10 results
- Languages All
Results 21 - 29 of 29 for RSQRT (0.08 sec)
-
tensorflow/compiler/mlir/lite/tests/ops.mlir
^bb0(%arg0: tensor<? x f32>): // CHECK: "tfl.rsqrt"(%arg0) %0 = "tfl.rsqrt"(%arg0): (tensor<? x f32>) -> tensor<? x f32> func.return %0 : tensor<? x f32> } // CHECK-LABEL: testRsqrtQuant func.func @testRsqrtQuant(%arg0: tensor<1x80x1x!quant.uniform<i8:f32, 0.048358432948589325:-128>>) -> tensor<1x80x1x!quant.uniform<i8:f32, 0.0066055487841367722:-128>> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 189.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
explicit FusedBatchNormV3Pat(::mlir::MLIRContext *context) : ::mlir::RewritePattern( "tf.FusedBatchNormV3", 1, context, {"tf.Add", "tf.Const", "tf.Mul", "tf.Rsqrt", "tf.Sub"}) {} ::mlir::LogicalResult matchAndRewrite( ::mlir::Operation *fused_batch_norm, ::mlir::PatternRewriter &rewriter) const override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
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/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/lite/ir/tfl_ops.td
); let results = (outs TFL_TensorOf<[F32, I32, I64, QI16, QUI8, TFL_Quint8]>:$output ); let hasOptions = 1; } def TFL_RsqrtOp: TFL_Op<"rsqrt", [Pure, QuantizableResult, TFL_SameFirstOperandAndFirstResultElementType, SameOperandsAndResultShape]> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
}]; } def TF_RsqrtGradOp : TF_Op<"RsqrtGrad", [Pure, TF_SameOperandsAndResultTypeResolveRef]> { let summary = "Computes the gradient for the rsqrt of `x` wrt its input."; let description = [{ Specifically, `grad = dy * -0.5 * y^3`, where `y = rsqrt(x)`, and `dy` is the corresponding input gradient. }]; let arguments = (ins TF_FpOrComplexTensor:$y, TF_FpOrComplexTensor:$dy
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K 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) -
RELEASE.md
* `TfLiteSignatureRunnerResizeInputTensor` * New C API function `TfLiteExtensionApisVersion` added to `tensorflow/lite/c/c_api.h`. * Add int8 and int16x8 support for RSQRT operator * Android NDK r25 is supported. ### Bug Fixes and Other Changes * Add TensorFlow Quantizer to TensorFlow pip package.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 730.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/schema_generated.h
"SLICE", "SIN", "TRANSPOSE_CONV", "SPARSE_TO_DENSE", "TILE", "EXPAND_DIMS", "EQUAL", "NOT_EQUAL", "LOG", "SUM", "SQRT", "RSQRT", "SHAPE", "POW", "ARG_MIN", "FAKE_QUANT", "REDUCE_PROD", "REDUCE_MAX", "PACK", "LOGICAL_OR", "ONE_HOT", "LOGICAL_AND", "LOGICAL_NOT",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 1M bytes - Viewed (0)