Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for createOutputReshapeOpForDynamic (0.28 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc

      std::vector<std::tuple<int64_t, int64_t>> lhs_out;
      std::vector<std::tuple<int64_t, int64_t>> rhs_out;
      std::vector<std::tuple<int64_t, int64_t, int64_t>> lhs_rhs_out;
    };
    
    TF::ReshapeOp createOutputReshapeOpForDynamic(
        Value value, ArrayRef<int64_t> shape, Value org_lhs, Value org_rhs,
        EinsumDimensionNumbers& dnums, Location loc, PatternRewriter* rewriter) {
      BoolAttr true_attr = rewriter->getBoolAttr(true);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 33.3K bytes
    - Viewed (0)
Back to top