Searched defs:axis_map (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/glsl/ |
| H A D | indexing_utils.h | 189 const ivec4 axis_map, in tidx_to_pos()
212 const ivec4 axis_map, in tidx_to_posi()
218 ivec3 lpos_to_pos(const ivec3 lpos, const ivec4 axis_map) { in lpos_to_pos()
230 #define load_texel_lpos(im, lpos, axis_map) \ argument
234 #define load_texel_lpos(im, lpos, axis_map) \ argument
242 #define write_texel_lpos(im, lpos, texel, axis_map) \ argument
246 #define write_texel_lpos(im, lpos, texel, axis_map) \ argument
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/api/containers/ |
| H A D | Tensor.cpp | 166 const std::vector<int64_t>& axis_map, in calculate_image_extents()
279 const std::vector<int64_t>& axis_map, in vTensorStorage()
|
| H A D | Tensor.h | 428 inline const std::vector<int64_t>& axis_map() const { in axis_map() function
|