Searched full:dim_order_alias (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torchgen/executorch/ |
| H A D | parse.py | 24 ET_FIELDS = ["kernels", "type_alias", "dim_order_alias"]
32 ei: Dict keys {kernels, type_alias, dim_order_alias}
40 dim_order_alias: dict[str, list[str]] = e.pop("dim_order_alias", {}) # type: ignore[assignment]
41 dim_order_alias.pop("__line__", None)
64 … else ETKernelKey.gen_from_yaml(arg_meta, type_alias, dim_order_alias) # type: ignore[arg-type]
|
| /aosp_15_r20/external/pytorch/tools/test/ |
| H A D | test_executorch_gen.py | 94 dim_order_alias:
124 dim_order_alias:
192 self.assertFalse({"kernels", "type_alias", "dim_order_alias"} < e.keys())
279 self.assertTrue({"kernels", "type_alias", "dim_order_alias"} < e.keys())
|
| /aosp_15_r20/external/executorch/docs/source/ |
| H A D | kernel-library-custom-aten-kernel.md | 74 * `dim_order_alias`(optional): similar to `type_alias`, we are giving names to possible dim order o…
97 dim_order_alias:
|
| /aosp_15_r20/external/executorch/kernels/ |
| H A D | README.md | 129 dim_order_alias:
152 * `dim_order_alias`(optional): similar to `type_alias`, we are giving names to possible dim order o…
|
| /aosp_15_r20/external/executorch/runtime/kernel/test/ |
| H A D | functions.yaml | 6 dim_order_alias:
|