Search Options

Results per page
Sort
Preferred Languages
Advance

Results 71 - 80 of 235 for legalize (0.13 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc

     public:
      explicit TflToStablehloPass() : PassWrapper() {}
      StringRef getArgument() const final { return "tfl-parse-stablehlo-ops"; }
      StringRef getDescription() const final {
        return "This pass will legalize TFLite custom Ops to StableHLO ops.";
      }
    
     private:
      void runOnOperation() override;
    
      void getDependentDialects(DialectRegistry& registry) const override {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jan 24 06:08:43 UTC 2024
    - 10.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_tf_xla_call_module_to_stablehlo_pass.cc

                             OperationPass<ModuleOp>> {
     public:
      StringRef getArgument() const final {
        return "tf-xla-call-module-op-to-stablehlo-pass";
      }
      StringRef getDescription() const final {
        return "Legalize TF_XlaCallModule Op to stablehlo";
      }
      void getDependentDialects(DialectRegistry &registry) const override {
        registry.insert<stablehlo::StablehloDialect, vhlo::VhloDialect,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jan 25 09:43:18 UTC 2024
    - 10.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc

      // `lower_static_tensor_list` pass.
      return element_type->isF32() || element_type->isInteger(64) ||
             element_type->isInteger(32) || element_type->isInteger(1);
    }
    
    // Only legalize TensorFlow TensorList ops if all TensorList ops are supported
    // natively.
    class LegalizeTensorListPass
        : public impl::LegalizeTensorListPassBase<LegalizeTensorListPass> {
     public:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 10.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/transforms/legalize_patterns.td

      [TF_UnsortedSegmentMaxOp, TFL_UnsortedSegmentMaxOp],
      [TF_UnsortedSegmentMinOp, TFL_UnsortedSegmentMinOp],
      [TF_UnsortedSegmentProdOp, TFL_UnsortedSegmentProdOp]] in {
      def Legalize#UnsortedSegmentOp[0] :
         Pat<(UnsortedSegmentOp[0] $data, $segment_ids, $num_segments),
                  (UnsortedSegmentOp[1] $data, (CreateTFCastToInt32Op $segment_ids),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 04 13:30:42 UTC 2024
    - 28.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/stablehlo/tests/tfl_legalize_hlo.mlir

    // COM: This file is there to check that the `tfl-legalize-hlo` pass exists in `odml-to-stablehlo-opt`.
    
    // RUN: odml-to-stablehlo-opt %s -tfl-legalize-hlo -split-input-file | FileCheck %s --dump-input=fail
    
    func.func @main(%arg0: tensor<5x7xf32>) -> tensor<5x7xf32> {
      func.return %arg0: tensor<5x7xf32>
    // CHECK-LABEL: main
    // CHECK: return %arg0 : tensor<5x7xf32>
    }
    
    // - transpose
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 40.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

            /*padding=*/state->padding,
            /*stride_h=*/state->stride_height,
            /*stride_w=*/state->stride_width);
      }
    
     private:
      // Legalize the given filter by converting it from TensorFlow filter data
      // format HWIO to TFLite Conv2D op filter data format OHWI and return Value
      // for the converted filter.  Requires that filter is verified by the match
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/stablehlo/tests/bridge/convert-tf-quant-types.mlir

    }
    
    // -----
    
    func.func @tf_const_invalid_proto() -> tensor<2x!tf_type.qint32> {
      // expected-error@+2 {{failed to get DenseElementAttr}}
      // expected-error@+1 {{failed to legalize operation 'tf.Const'}}
      %0 = "tf.Const"() { value = #tf_type<tensor_proto : "0x2532"> : tensor<2x!tf_type.qint32> } : () -> tensor<2x!tf_type.qint32>
      func.return %0 :  tensor<2x!tf_type.qint32>
    }
    
    // -----
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 25.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/passes.h

    // (either defined in TensorArrayV3 or implied in the first write).
    std::unique_ptr<OperationPass<ModuleOp>>
    CreateTensorArrayOpsDecompositionPass();
    
    // Create a pass that legalize TFG to TF dialect.
    std::unique_ptr<Pass> CreateLegalizeTFGToTFEPass();
    
    // Matches sequence of ops to TensorFlow fused kernels. This pass should not be
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 31.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc

                            lower_to_xla_hlo);
    
      if (lower_to_xla_hlo) {
        // This pass operates on MHLO control flow ops so it should be legalized
        // after the control flow ops are legalized.
        pm.addPass(mlir::mhlo::CreateLegalizeTFCommunicationPass());
    
        // Everything should be MHLO after this.
        if (!allow_partial_conversion) {
          pm.addNestedPass<mlir::func::FuncOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 17:24:39 UTC 2024
    - 45.3K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tf2xla/transforms/passes.h

        bool prefer_tf2xla = false);
    
    // Legalizes TF/XLA communication ops (TF dialect) to HLO dialect communication
    // ops.
    std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeTFCommunicationPass();
    
    // Legalizes TF/XLA collective ops (TF dialect) to HLO dialect collective
    // ops.
    std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeTFCollectivePass();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 4.8K bytes
    - Viewed (0)
Back to top