OutputTensors

Function Usage

Sets an output tensor.

Prototype

1
OpTilingContextBuilder &OutputTensors(const std::vector<gert::Tensor *> &outputs)

Parameters

Parameter

Input/Output

Description

outputs

Input

Pointer array of output tensors.

Returns

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

Constraints

  • OutputTensors 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.