InputTensors

Function Usage

Sets the input tensor pointer. During tiling computation, the input tensor pointer can be obtained from the TilingContext built through the Builder class.

Prototype

1

OpTilingContextBuilder &InputTensors(const std::vector<gert::Tensor *> &inputs)

Parameters

Returns

A reference to the OpTilingContextBuilder object. It is used for chaining method calls.

Constraints

• InputTensors must be set before the Build method call. Otherwise, the constructed TilingContext will contain undefined data.
• The caller has the memory ownership of the void* parameter passed through a pointer. They must ensure that the pointer is valid throughout the lifecycle of the ContextHolder object.