LLVM: gen-regalloc-priority-test-model Namespace Reference (original) (raw)
| Functions |
|---|
| Variables | |
|---|---|
| str | POLICY_DECISION_LABEL = "priority" |
| str | POLICY_OUTPUT_SPEC |
| list | PER_LIVEINTERVAL_INT64_FEATURE_LIST = ["li_size", "stage"] |
| list | PER_LIVEINTERVAL_FLOAT32_FEATURE_LIST = ["weight"] |
| tuple | PER_LIVEINTERVAL_FEATURE_LIST |
| tuple | CONTEXT_FEATURE_LIST = ("discount", "reward", "step_type") |
◆ build_mock_model()
| gen-regalloc-priority-test-model.build_mock_model | ( | path | ) |
|---|
◆ get_input_signature()
| gen-regalloc-priority-test-model.get_input_signature | ( | ) |
|---|
◆ get_output_spec_path()
| gen-regalloc-priority-test-model.get_output_spec_path | ( | path | ) |
|---|
◆ main()
| gen-regalloc-priority-test-model.main | ( | argv | ) |
|---|
◆ CONTEXT_FEATURE_LIST
tuple gen-regalloc-priority-test-model.CONTEXT_FEATURE_LIST = ("discount", "reward", "step_type")
◆ PER_LIVEINTERVAL_FEATURE_LIST
tuple gen-regalloc-priority-test-model.PER_LIVEINTERVAL_FEATURE_LIST
◆ PER_LIVEINTERVAL_FLOAT32_FEATURE_LIST
list gen-regalloc-priority-test-model.PER_LIVEINTERVAL_FLOAT32_FEATURE_LIST = ["weight"]
◆ PER_LIVEINTERVAL_INT64_FEATURE_LIST
list gen-regalloc-priority-test-model.PER_LIVEINTERVAL_INT64_FEATURE_LIST = ["li_size", "stage"]
◆ POLICY_DECISION_LABEL
str gen-regalloc-priority-test-model.POLICY_DECISION_LABEL = "priority"
◆ POLICY_OUTPUT_SPEC
str gen-regalloc-priority-test-model.POLICY_OUTPUT_SPEC
Initial value:
1= """
2[
3 {
4 "logging_name": "priority",
5 "tensor_spec": {
6 "name": "StatefulPartitionedCall",
7 "port": 0,
8 "type": "float",
9 "shape": [
10 1
11 ]
12 }
13 }
14]
15"""
Definition at line 12 of file gen-regalloc-priority-test-model.py.