- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for seq_dim (0.09 sec)
-
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) -
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 (1)