tvm::topi::generic Namespace Reference

Functions
Schedule	default_schedule (const Target &target, const Array< Tensor > &outs)
	Create a generic default schedule for the given output tensors. More...
Schedule	default_schedule_auto_inline (const Target &target, const Array< Tensor > &outs)
	Create a generic default schedule for the given output tensors, and apply auto inline. More...
Schedule	schedule_extern (const Target &target, const Array< Tensor > &outs)
	Schedule an extern op followed by injective operations. 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 a generic schedule for the given injective ops. More...

Function Documentation

default_schedule()

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

inline

Create a generic default schedule for the given output tensors.

Parameters

target	The target to generate a schedule for.
outs	The output tensors.

Returns

A schedule for the given ops.

default_schedule_auto_inline()

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

inline

Create a generic default schedule for the given output tensors, and apply auto inline.

Parameters

target	The target to generate a schedule for.
outs	The output tensors.

Returns

A schedule for the given ops.

schedule_extern()

Schedule tvm::topi::generic::schedule_extern ( const Target &  target, const Array< Tensor > &  outs  )

inline

Schedule an extern op followed by injective operations.

Parameters

target	The target to generate a schedule for.
outs	The output tensors.

Returns

A schedule for the op.

schedule_injective()

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

inline

Create a generic schedule for the given injective ops.

Parameters

target	The target to generate a schedule for.
outs	The output tensors.

Returns

A schedule for the given ops.

schedule_injective_from_existing()

Schedule tvm::topi::generic::schedule_injective_from_existing ( Schedule  sch, const Tensor &  out  )

inline

Updates an existing schedule for the given injective ops.

Parameters

sch	The schedule to update.
out	The tensor representing the injective op.

Returns

The updated schedule.