FeedsRepeat

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	×
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×
Kirin X90 处理器系列产品	√
Kirin 9030 处理器系列产品	√

功能说明

• 算子功能：

  对于输入feeds，根据输入feeds_repeat_times,将对应的feeds的第0维上的数据复制对应的次数，并将输出y的第0维padding到output_feeds_size的大小

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
feeds	输入	等待处理的对象。	BFLOAT16、FLOAT16、FLOAT32	ND
feeds_repeat_times	输入	描述对feeds每行的repeat次数。	INT32、INT64	ND
y	输出	repeat完成后的张量，dtype与feeds保持一致。	BFLOAT16、FLOAT16、FLOAT32	ND
output_feeds_size	属性	输出的第0维的数值等于该数值。	INT

• Kirin X90/Kirin 9030 处理器系列产品: 不支持BFLOAT16。

约束说明

在算子实现过程中，feeds_repeat_times需搬入ub并做精度转换和累加，故feeds_repeat_times数据规模不能超过ub大小的四分之一。

调用说明

调用方式	调用样例	说明
geir接口	test_geir_feeds_repeat	通过feeds_repeat_proto接口方式调用feeds_repeat算子。