Home
last modified time | relevance | path

Searched refs:unhash_axis_map (Results 1 – 14 of 14) sorted by relevance

/external/executorch/backends/vulkan/runtime/graph/ops/glsl/
Dembedding.glsl27 const lowp ivec4 out_axis_map = unhash_axis_map(out_layout);
31 const lowp ivec4 in_axis_map = unhash_axis_map(in_layout);
34 const lowp ivec4 weight_axis_map = unhash_axis_map(weight_layout);
Dbinary_op.glsl34 const lowp ivec4 out_axis_map = unhash_axis_map(out_layout);
38 const lowp ivec4 in_axis_map = unhash_axis_map(in_layout);
41 const lowp ivec4 other_axis_map = unhash_axis_map(other_layout);
Dcopy_offset.glsl27 const lowp ivec4 out_axis_map = unhash_axis_map(out_layout);
30 const lowp ivec4 in_axis_map = unhash_axis_map(in_layout);
Drepeat_interleave.glsl28 const lowp ivec4 tout_axis_map = unhash_axis_map(tout_layout);
31 const lowp ivec4 tin_axis_map = unhash_axis_map(tin_layout);
Dconv1d.glsl36 const lowp ivec4 out_axis_map = unhash_axis_map(out_layout);
39 const lowp ivec4 in_axis_map = unhash_axis_map(in_layout);
42 const lowp ivec4 kernel_axis_map = unhash_axis_map(kernel_layout);
45 const lowp ivec4 bias_axis_map = unhash_axis_map(bias_layout);
Daddmm_naive_texture3d.glsl37 const lowp ivec4 out_axis_map = unhash_axis_map(out_layout);
41 const lowp ivec4 mat1_axis_map = unhash_axis_map(mat1_layout);
45 const lowp ivec4 mat2_axis_map = unhash_axis_map(mat2_layout);
50 const lowp ivec4 bias_axis_map = unhash_axis_map(bias_layout);
Daddmm_optimized.glsl39 const lowp ivec4 out_axis_map = unhash_axis_map(out_layout);
43 const lowp ivec4 mat1_axis_map = unhash_axis_map(mat1_layout);
46 const lowp ivec4 mat2_axis_map = unhash_axis_map(mat2_layout);
50 const lowp ivec4 bias_axis_map = unhash_axis_map(bias_layout);
Dcopy_channel_offset.glsl40 const lowp ivec4 out_axis_map = unhash_axis_map(out_layout);
44 const lowp ivec4 in_axis_map = unhash_axis_map(in_layout);
Dnative_layer_norm.glsl35 const lowp ivec4 in_axis_map = unhash_axis_map(in_layout);
39 const lowp ivec4 out_axis_map = unhash_axis_map(out_layout);
Dnchw_to_image.glsl33 const lowp ivec4 axis_map = unhash_axis_map(t_layout);
Dimage_to_nchw.glsl33 const lowp ivec4 axis_map = unhash_axis_map(t_layout);
Dbitw8_image_to_nchw_nobitw8buffer.glsl29 const lowp ivec4 axis_map = unhash_axis_map(t_layout);
Dnchw_to_bitw8_image_nobitw8buffer.glsl30 const lowp ivec4 axis_map = unhash_axis_map(t_layout);
Dindexing_utils.h258 #define unhash_axis_map(hash) \ macro