Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 25 of 25 for RSQRT (0.19 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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