- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for BuildListOp (0.09 sec)
-
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) -
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) -
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)