Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for ConvertTransposeToDecreaseRank (0.29 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/lite/ir/tfl_canonicalize.td 

    // %1 = "tfl.transpose"(%0, %cst_0) : (tensor<56x8x56x7xf32>, tensor<4xi32>) -> tensor<8x56x56x7xf32>
// %2 = "tfl.reshape"(%1, %cst_1) : (tensor<8x56x56x7xf32>, tensor<7xi32>) -> tensor<1x1x8x56x56x7x1xf32>
def ConvertTransposeToDecreaseRank : Pat<
  (TFL_TransposeOp:$output_transpose $input, (Arith_ConstantOp:$permutation $_)),
  (TFL_ReshapeOp
    (TFL_TransposeOp
      (TFL_ReshapeOp $input, (Arith_ConstantOp (GetSqueezedShape $input))),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Dec 13 20:41:03 UTC 2023
    - 2.7K bytes
    - Viewed (0)
Back to top