Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for perm_values (0.09 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td 

    def FuseTransposeIntoBatchMatMulLHS: Pat<
  (TFL_BatchMatMulOp
    (TFL_TransposeOp:$transposed_value $input, (Arith_ConstantOp:$perm_value $p0)),
    $rhs, $adj_x, $adj_y, $asymmetric_quantize_inputs),
  (TFL_BatchMatMulOp $input, $rhs, ConstBoolAttrTrue, $adj_y, $asymmetric_quantize_inputs),
  [(AreLastTwoDimsTransposed $perm_value),
   (IsBoolAttrEqual<"false"> $adj_x)]>;

// Fuses TFL_TransposeOp after TFL_BatchMatMulOp into the BMM.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 20:31:41 UTC 2024
    - 66.4K bytes
    - Viewed (0)
Back to top