Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for ConvertTFEinsumOp (0.43 sec)

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

    // cases, we can convert this op to other TF Ops, which in later passes
    // properly convert to TF Lite ops.
    struct ConvertTFEinsumOp : public OpRewritePattern<TF::EinsumOp> {
     public:
      explicit ConvertTFEinsumOp(MLIRContext* context)
          : OpRewritePattern<TF::EinsumOp>(context) {}
    
      LogicalResult matchAndRewrite(TF::EinsumOp op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Dec 12 02:01:03 UTC 2020
    - 2.1K bytes
    - Viewed (0)
Back to top