SetValueOperation

功能

实现将输入src中的内容拷贝到输入dst指定位置中的功能。

例如:dst: (a,b,c), src: (d,e,f)

dst[starts[0]: ends[0], starts[1]: ends[1], starts[2]: ends[2]] = src

其中,starts[0]: ends[0]长度为src第0维的维度大小,starts[1]: ends[1]长度为src第1维的维度大小,starts[2]: ends[2]长度为src第2维的维度大小。

约束

定义

struct SetValueParam {
     SVector<int64_t> starts;
     SVector<int64_t> ends;
     SVector<int64_t> strides;
};

成员

成员名称

描述

starts

每个轴的start。

ends

每个轴的end。

strides

拷贝时步长,当前仅支持strides为全1。

输入

参数

维度

数据类型

格式

src

[-1,…,-1]

-1表示当前维度的大小没有约束。

float16/float/int32/int64

ND

dst

[-1,…,-1]

-1表示当前维度的大小没有约束。

float16/float/int32/int64

ND

输出

参数

维度

数据类型

格式

dst

[-1,…,-1]

-1表示当前维度的大小没有约束。

float16/float/int32/int64

ND

输出dst与同名输入指向同一地址,即进行原地修改。