Skip to content

Upd constant of shape

Maxence Naud requested to merge upd_ConstantOfShape into dev

Context

Use Tensor as input of the ConstantOfShape kernel to avoid redundant size computation.

Kernel performance is unchanged. (x-axis is the number of elements in the output)

ConstantOfShape_forward_time_comparison_with_speedup

Edited by Maxence Naud

Merge request reports

Loading