Search Options

Results per page
Sort
Preferred Languages
Advance

Results 101 - 102 of 102 for 1xi64 (0.03 sec)

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

          while (auto cast_op =
                     llvm::dyn_cast_or_null<CastOp>(value.getDefiningOp())) {
            // Consider cast compatibility in case
            //    %cast = "tf.Cast"(%0) : (tensor<2xi64>) -> tensor<2xf32>
            // is skipped.
            if (cast_op.getSrcT() != cast_op.getDstT()) {
              break;
            }
            value = cast_op.getOperand();
          }
          return value;
        };
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 28.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

    // For example,
    //
    //   %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,
    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