ks.c.j2 712 字节
Newer Older
1
{% from "operator_utils.c.j2" import name_map, register_name_map, register_base_kernel_name %}
2
// this file is generated by paddle/phi/api/yaml/generator/generate_op.py, do not edit.
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
#include "paddle/phi/core/compat/op_utils.h"
#include "paddle/utils/small_vector.h"

namespace phi {

{% for api in apis %}
  {% if api is base_api %}
{{name_map(api)}}
  {% endif %}
{% endfor %}
{% for api in backward_apis %}
  {% if api is base_api %}
{{name_map(api)}}
  {% endif %}
{% endfor %}
}  // namespace phi

{% for api in apis + backward_apis %}
21 22 23
  {% if api["name"] != api["op_name"] %}
{{register_base_kernel_name(api)}}
  {% endif %}
24 25 26 27
  {% if api is base_api %}
{{register_name_map(api)}}
  {% endif %}
{% endfor %}