Help Center/ Atlas 300 Application (Model 3000)/ TE API Reference/ Build APIs/ te.lang.cce.auto_schedule(outs)
Updated on 2022-03-13 GMT+08:00
View PDF
Share

te.lang.cce.auto_schedule(outs)

Generates the schedule of DSL. This API is defined in cce.py.

Parameter Description

outs: computational graph description of the operator, that is, DSL

Return Value

schedule: computation schedule of the operator

Calling Example

import te.lang.cce 
from te import tvm 
import topi

shape = (28,28)
dtype = "float16"
# Define input.
data = tvm.placeholder(shape, name="data", dtype=dtype)
# Describe the computation process of the operator.
res = te.lang.cce.vabs(data)
with tvm.target.cce():
    # Generate the schedule object.
      sch = topi.generic.auto_schedule(res)
Parent topic: Build APIs