tvm
call.h
Go to the documentation of this file.
1 /*
2  * Licensed to the Apache Software Foundation (ASF) under one
3  * or more contributor license agreements. See the NOTICE file
4  * distributed with this work for additional information
5  * regarding copyright ownership. The ASF licenses this file
6  * to you under the Apache License, Version 2.0 (the
7  * "License"); you may not use this file except in compliance
8  * with the License. You may obtain a copy of the License at
9  *
10  * http://www.apache.org/licenses/LICENSE-2.0
11  *
12  * Unless required by applicable law or agreed to in writing,
13  * software distributed under the License is distributed on an
14  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15  * KIND, either express or implied. See the License for the
16  * specific language governing permissions and limitations
17  * under the License.
18  */
19 
24 #ifndef TVM_RELAY_ATTRS_CALL_H_
25 #define TVM_RELAY_ATTRS_CALL_H_
26 
27 #include <tvm/ir/attrs.h>
28 
29 #include <string>
30 
31 namespace tvm {
32 namespace relay {
33 
37 struct CallLoweredAttrs : public tvm::AttrsNode<CallLoweredAttrs> {
40 
41  TVM_DECLARE_ATTRS(CallLoweredAttrs, "relay.attrs.CallLoweredAttrs") {
43  .describe("Metadata attached to the lowered function call.")
44  .set_default(Map<String, ObjectRef>());
45  }
46 };
47 
48 } // namespace relay
49 } // namespace tvm
50 #endif // TVM_RELAY_ATTRS_CALL_H_
The base class of the all the Use "curiously recurring template pattern".
Definition: attrs.h:870
Map container of NodeRef->NodeRef in DSL graph. Map implements copy on write semantics,...
Definition: map.h:1271
Helpers for attribute objects.
#define TVM_ATTR_FIELD(FieldName)
Declare an attribute field.
Definition: attrs.h:76
runtime implementation for LibTorch/TorchScript.
Definition: analyzer.h:36
Metadata for calls to TIR functions, useful for program analysis crossing Relay and TIR.
Definition: call.h:37
Map< String, ObjectRef > metadata
Additional metadata attached to the call node. Should be replaced by explict fields.
Definition: call.h:39
TVM_DECLARE_ATTRS(CallLoweredAttrs, "relay.attrs.CallLoweredAttrs")
Definition: call.h:41