Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for seq_dim (0.16 sec)

  1. tensorflow/compiler/mlir/lite/transforms/legalize_patterns.td

              ConstantAttr<RankedI64ElementsAttr<[]>,"static_cast<int64_t>(0)">))>;
    def LegalizeReverseSequence : Pat<
      (TF_ReverseSequenceOp $input, $seq_lengths, $seq_dim, $batch_dim),
      (TFL_ReverseSequenceOp $input, $seq_lengths,
          (convertIntAttrTo32Bit $seq_dim), (convertIntAttrTo32Bit $batch_dim))>;
    def LegalizeRound : Pat<(TF_RoundOp $arg), (TFL_RoundOp $arg)>;
    def LegalizeRsqrt : Pat<(TF_RsqrtOp $arg), (TFL_RsqrtOp $arg)>;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 04 13:30:42 UTC 2024
    - 28.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-with-tf2xla-hlo-importer.mlir

      func.func @reverse_sequence(%arg0: tensor<4x2x3x1x1xi32>, %arg1: tensor<3xi32>) -> tensor<4x2x3x1x1xi32> {
        // CHECK-NOT: tf.ReverseSequence
        %0 = "tf.ReverseSequence"(%arg0, %arg1) {batch_dim = 2 : i64, seq_dim = 0 : i64}: (tensor<4x2x3x1x1xi32>, tensor<3xi32>) -> tensor<4x2x3x1x1xi32>
        func.return %0 : tensor<4x2x3x1x1xi32>
      }
    
      // CHECK-LABEL: mirror_pad
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 06 15:32:52 UTC 2024
    - 38.6K bytes
    - Viewed (0)
Back to top