xla
[Core Aten Op set] Codegen topk
#6244
Open

[Core Aten Op set] Codegen topk #6244

ManfeiBai wants to merge 9 commits into master from ManfeiBai-patch-51
ManfeiBai
ManfeiBai Update xla_native_functions.yaml
ba1895a7
ManfeiBai Update ops_xla_shape_fn.h
3a0cda14
ManfeiBai Update ops_xla_shape_fn.cpp
ee03ca6e
ManfeiBai Update aten_xla_type.cpp
e7480721
ManfeiBai Delete torch_xla/csrc/ops/topk.h
63bbe1d5
ManfeiBai Delete torch_xla/csrc/ops/topk.cpp
238dc7fc
ManfeiBai Update ops_lower_fn.cpp
c0328a42
ManfeiBai Update ops_xla_shape_fn.cpp
e129294f
ManfeiBai Update ops_xla_shape_fn.h
fa2dffa7
ManfeiBai ManfeiBai closed this 1 year ago
ManfeiBai ManfeiBai reopened this 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone