Searched defs:dim_ref (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Slice.cpp | 39 ValueRef dim_ref, in add_slice_tensor_copy_node()
157 ValueRef dim_ref, in get_slice_sizes()
199 ValueRef dim_ref, in check_slice_view_args()
248 ValueRef dim_ref, in add_slice_view_node()
|
| H A D | IndexSelect.cpp | 103 int64_t get_dim_idx(ComputeGraph& graph, ValueRef in, ValueRef dim_ref) { in get_dim_idx()
112 ValueRef dim_ref = args[1]; in index_select() local
|
| H A D | Split.cpp | 101 ValueRef dim_ref, in add_split_with_sizes_default_node()
119 ValueRef dim_ref, in add_split_tensor_node()
|
| H A D | Unsqueeze.cpp | 20 ValueRef dim_ref, in add_unsqueeze_node()
|
| H A D | Select.cpp | 50 const ValueRef dim_ref, in add_select_int_node()
|
| H A D | Cat.cpp | 22 ValueRef dim_ref, in add_cat_default_node()
|