tvm
Functions
tvm::topi::x86 Namespace Reference

Functions

Schedule schedule_binarize_pack (const Target &target, const Array< Tensor > &outs)
 Create a generic schedule for binarize_pack. More...
 
Schedule schedule_binary_dense (const Target &target, const Array< Tensor > &outs)
 Create a generic schedule for binary_dense. More...
 
Schedule MakeDefaultSchedule (const Target &target, const Array< Tensor > &outs, bool auto_inline)
 Helper to create a default x86 schedule for the given ops. More...
 
Schedule default_schedule (const Target &target, const Array< Tensor > &outs)
 Create a default x86 schedule for the given ops. More...
 
Schedule default_schedule_auto_inline (const Target &target, const Array< Tensor > &outs)
 Create a default x86 schedule for the given ops, with auto inline. More...
 
Schedule schedule_injective_from_existing (Schedule sch, const Tensor &out)
 Updates an existing schedule for the given injective ops. More...
 
Schedule schedule_injective (const Target &target, const Array< Tensor > &outs)
 Create an x86 schedule for the given injective ops. More...
 

Function Documentation

◆ default_schedule()

Schedule tvm::topi::x86::default_schedule ( const Target target,
const Array< Tensor > &  outs 
)
inline

Create a default x86 schedule for the given ops.

Parameters
targetThe target to generate a schedule for.
outsThe output tensors.
Returns
A schedule for the given ops.

◆ default_schedule_auto_inline()

Schedule tvm::topi::x86::default_schedule_auto_inline ( const Target target,
const Array< Tensor > &  outs 
)
inline

Create a default x86 schedule for the given ops, with auto inline.

Parameters
targetThe target to generate a schedule for.
outsThe output tensors.
Returns
A schedule for the given ops.

◆ MakeDefaultSchedule()

Schedule tvm::topi::x86::MakeDefaultSchedule ( const Target target,
const Array< Tensor > &  outs,
bool  auto_inline 
)
inline

Helper to create a default x86 schedule for the given ops.

Parameters
targetThe target to generate a schedule for.
outsThe output tensors.
auto_inlineWhether to apply the auto inline step.
Returns
A schedule for the given ops.

◆ schedule_binarize_pack()

Schedule tvm::topi::x86::schedule_binarize_pack ( const Target target,
const Array< Tensor > &  outs 
)
inline

Create a generic schedule for binarize_pack.

Parameters
targetThe target to generate a schedule for.
outsThe output tensors.
Returns
A schedule for the given ops.

◆ schedule_binary_dense()

Schedule tvm::topi::x86::schedule_binary_dense ( const Target target,
const Array< Tensor > &  outs 
)
inline

Create a generic schedule for binary_dense.

Parameters
targetThe target to generate a schedule for.
outsThe output tensors.
Returns
A schedule for the given ops.

◆ schedule_injective()

Schedule tvm::topi::x86::schedule_injective ( const Target target,
const Array< Tensor > &  outs 
)
inline

Create an x86 schedule for the given injective ops.

Parameters
targetThe target to generate a schedule for.
outsThe output tensors.
Returns
A schedule for the given ops.

◆ schedule_injective_from_existing()

Schedule tvm::topi::x86::schedule_injective_from_existing ( Schedule  sch,
const Tensor out 
)
inline

Updates an existing schedule for the given injective ops.

Parameters
schThe schedule to update.
outThe tensor representing the injective op.
Returns
The updated schedule.