FilesLast commitLast update
9 months ago
2 months ago
15 days ago
9 months ago
15 days ago
10 months ago
19 days ago
README

FeedsRepeat

产品支持情况

产品 是否支持
Ascend 950PR&950DT系列产品 √
Atlas A3系列产品 √
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取值在Atlas A2/A3系列产品不能超过48 kb, 在Ascend 950PR&950DT系列产品系列产品不能超过64 kb。

调用说明

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