Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 29 of 29 for RSQRT (0.08 sec)

  1. 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)
  2. 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)
  3. tensorflow/compiler/jit/mark_for_compilation_pass.cc

                "Floor", "IsFinite", "IsInf", "IsNan", "Inv", "Reciprocal", "Log",
                "Log1p", "Invert", "LogicalNot", "Ndtri", "Neg", "Rint", "Round",
                "Rsqrt", "Sigmoid", "Sign", "Sinh", "Softplus", "Softsign", "Sqrt",
                "Square", "Tan", "Tanh", "Real", "Imag", "Erf", "Erfc", "Erfinv",
                "Lgamma", "Digamma",
                // Binary
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. tensorflow/compiler/mlir/lite/schema/schema_generated.h

    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 1M bytes
    - Viewed (0)
Back to top