We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e9fd7ef commit 55e869bCopy full SHA for 55e869b
1 file changed
paddle/cinn/hlir/framework/op_lowering_impl.cc
@@ -45,8 +45,8 @@ using cinn::common::Type;
45
using cinn::hlir::op::ExternalApiRegistry;
46
47
// Collect the temporary tensors from a computational graph.
48
-std::vector<ir::Buffer> GetTempBuffers(const std::vector<Tensor>& tensor_args,
49
- Expr body) {
+std::vector<ir::Buffer> GetTempBuffers(
+ const std::vector<cin::ir::Tensor>& tensor_args, Expr body) {
50
std::unordered_set<std::string> tensor_arg_names;
51
std::unordered_set<std::string> buffer_arg_names;
52
for (auto& tensor : tensor_args) {
0 commit comments