Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for ConvertPackToReshape (0.2 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

    //   %0 = tf.Pack(%input) {axis = 0} // %input : tensor<2x3xf32>
    //
    // can be canonicalized to
    //
    //   %shape = "tf.Const"() {value = dense<[1, 2, 3]> : tensor<3xi64>}
    //   %0 = tf.Reshape(%input, %shape)
    struct ConvertPackToReshape : public OpRewritePattern<PackOp> {
      using OpRewritePattern<PackOp>::OpRewritePattern;
    
      LogicalResult matchAndRewrite(PackOp pack_op,
                                    PatternRewriter &rewriter) const override {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
Back to top