Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for BuildListOp (0.09 sec)

  1. tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc

            gl_op, rewriter.getIndexAttr(num_tensors));
        return success();
      }
    };
    
    class BuildConstantListAsAttr : public OpRewritePattern<BuildListOp> {
      using OpRewritePattern<BuildListOp>::OpRewritePattern;
    
     public:
      LogicalResult matchAndRewrite(BuildListOp bl_op,
                                    PatternRewriter &rewriter) const override {
        SmallVector<Attribute, 4> array_list;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 21 16:55:41 UTC 2023
    - 38.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc

        // values with tensor types via the tfr.cast ops. These ops will be fused
        // to the current op as well, so all the tfr.cast op inputs will be inputs
        // to the result op.
        if (auto list_op = dyn_cast_or_null<BuildListOp>(input_op)) {
          // Find out all the inputs to the build list op
          // TODO(fengliuai): make build_list op only take tensor argument
          llvm::SmallVector<Attribute, 4> list_input_types;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfr/passes/decompose.cc

                  op->getLoc(), unconstrainted_tensor_type, op->getOperand(i));
              variadic_operands.push_back(casted);
            }
            auto build_list_op = builder.create<BuildListOp>(
                op->getLoc(), list_type, variadic_operands);
            new_operands.push_back(build_list_op.getOut());
          } else {
            auto attr_name = compose_func.getArgAttrOfType<StringAttr>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
Back to top