Home
last modified time | relevance | path

Searched refs:layout_declare_tensor (Results 1 – 25 of 48) sorted by relevance

12

/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/glsl/
H A Dnative_layer_norm.glsl20 ${layout_declare_tensor(B, "w", "t_out", DTYPE, STORAGE)}
21 ${layout_declare_tensor(B, "w", "t_mean", DTYPE, STORAGE)}
22 ${layout_declare_tensor(B, "w", "t_rstd", DTYPE, STORAGE)}
24 ${layout_declare_tensor(B, "r", "t_in", DTYPE, STORAGE)}
25 ${layout_declare_tensor(B, "r", "t_weight", DTYPE, STORAGE)}
26 ${layout_declare_tensor(B, "r", "t_bias", DTYPE, STORAGE)}
H A Drotary_embedding.glsl19 ${layout_declare_tensor(B, "w", "xqout", DTYPE, STORAGE)}
20 ${layout_declare_tensor(B, "w", "xkout", DTYPE, STORAGE)}
21 ${layout_declare_tensor(B, "r", "xq", DTYPE, STORAGE)}
22 ${layout_declare_tensor(B, "r", "xk", DTYPE, STORAGE)}
23 ${layout_declare_tensor(B, "r", "freqs_cos", DTYPE, STORAGE)}
24 ${layout_declare_tensor(B, "r", "freqs_sin", DTYPE, STORAGE)}
H A Dconv2d_dw.glsl21 ${layout_declare_tensor(0, "w", "t_out", DTYPE, "texture3d")}
22 ${layout_declare_tensor(1, "r", "t_in", DTYPE, "texture3d")}
23 ${layout_declare_tensor(2, "r", "t_kernel", DTYPE, "texture2d")}
24 ${layout_declare_tensor(3, "r", "t_bias", DTYPE, "texture2d")}
H A Dconv2d_dw_output_tile.glsl23 ${layout_declare_tensor(0, "w", "t_out", DTYPE, "texture3d")}
24 ${layout_declare_tensor(1, "r", "t_in", DTYPE, "texture3d")}
25 ${layout_declare_tensor(2, "r", "t_kernel", DTYPE, "texture2d")}
26 ${layout_declare_tensor(3, "r", "t_bias", DTYPE, "texture2d")}
H A Dconv_transpose2d.glsl21 ${layout_declare_tensor(0, "w", "t_out", DTYPE, "texture3d")}
22 ${layout_declare_tensor(1, "r", "t_in", DTYPE, "texture3d")}
23 ${layout_declare_tensor(2, "r", "t_kernel", DTYPE, "texture2d")}
24 ${layout_declare_tensor(3, "r", "t_bias", DTYPE, "texture2d")}
H A Dq_8w_linear.glsl25 ${layout_declare_tensor(0, "w", "t_out", DTYPE, STORAGE)}
26 ${layout_declare_tensor(1, "r", "t_mat1", DTYPE, STORAGE)}
27 ${layout_declare_tensor(2, "r", "t_qmat2", "int8", STORAGE)}
28 ${layout_declare_tensor(3, "r", "t_scales", DTYPE, STORAGE)}
H A Dindex_select.glsl19 ${layout_declare_tensor(0, "w", "t_out", DTYPE, STORAGE)}
20 ${layout_declare_tensor(1, "r", "t_in", DTYPE, STORAGE)}
21 ${layout_declare_tensor(2, "r", "t_idx", "int", STORAGE)}
H A Dindex_select_channel.glsl19 ${layout_declare_tensor(0, "w", "t_out", DTYPE, STORAGE)}
20 ${layout_declare_tensor(1, "r", "t_in", DTYPE, STORAGE)}
21 ${layout_declare_tensor(2, "r", "t_idx", "int", STORAGE)}
H A Dq_4w_linear.glsl27 ${layout_declare_tensor(B, "w", "ret", DTYPE, STORAGE)}
28 ${layout_declare_tensor(B, "r", "x", DTYPE, STORAGE)}
29 ${layout_declare_tensor(B, "r", "weights", "uint8", "buffer")}
30 ${layout_declare_tensor(B, "r", "qparams", DTYPE, STORAGE)}
H A Dmatmul_naive_buffer.glsl19 ${layout_declare_tensor(B, "w", "t_out", DTYPE, "buffer")}
20 ${layout_declare_tensor(B, "r", "t_mat1", DTYPE, "buffer")}
21 ${layout_declare_tensor(B, "r", "t_mat2", DTYPE, "buffer")}
H A Dembedding.glsl17 ${layout_declare_tensor(B, "w", "t_out", DTYPE, STORAGE)}
18 ${layout_declare_tensor(B, "r", "t_in", "int", STORAGE)}
19 ${layout_declare_tensor(B, "r", "t_weight", DTYPE, STORAGE)}
H A Daddmm_naive_texture3d.glsl19 ${layout_declare_tensor(B, "w", "out_tensor", DTYPE, "texture3d")}
20 ${layout_declare_tensor(B, "r", "mat1_tensor", DTYPE, "texture3d")}
21 ${layout_declare_tensor(B, "r", "mat2_tensor", DTYPE, "texture3d")}
23 ${layout_declare_tensor(B, "r", "bias_tensor", DTYPE, "texture3d")}
H A Dconv1d.glsl19 ${layout_declare_tensor(B, "w", "t_out", DTYPE, STORAGE)}
20 ${layout_declare_tensor(B, "r", "t_in", DTYPE, STORAGE)}
21 ${layout_declare_tensor(B, "r", "kernel_in", DTYPE, STORAGE)}
22 ${layout_declare_tensor(B, "r", "bias_in", DTYPE, STORAGE)}
H A Dq_8w_linear_optimized.glsl48 ${layout_declare_tensor(0, "w", "t_out", DTYPE, STORAGE)}
49 ${layout_declare_tensor(1, "r", "t_mat1", DTYPE, STORAGE)}
50 ${layout_declare_tensor(2, "r", "t_qmat2", "int8", STORAGE)}
51 ${layout_declare_tensor(3, "r", "t_scales", DTYPE, STORAGE)}
H A Dconv2d.glsl21 ${layout_declare_tensor(0, "w", "t_out", DTYPE, "texture3d")}
22 ${layout_declare_tensor(1, "r", "t_in", DTYPE, "texture3d")}
23 ${layout_declare_tensor(2, "r", "t_kernel", DTYPE, "texture2d")}
24 ${layout_declare_tensor(3, "r", "t_bias", DTYPE, "texture2d")}
H A Dbinary_op.glsl19 ${layout_declare_tensor(B, "w", "t_out", DTYPE, STORAGE)}
20 ${layout_declare_tensor(B, "r", "t_in", DTYPE, STORAGE)}
21 ${layout_declare_tensor(B, "r", "t_other", DTYPE, STORAGE)}
H A Dconv2d_pw.glsl23 ${layout_declare_tensor(0, "w", "t_out", DTYPE, "texture3d")}
24 ${layout_declare_tensor(1, "r", "t_in", DTYPE, "texture3d")}
25 ${layout_declare_tensor(2, "r", "t_kernel", DTYPE, "texture2d")}
26 ${layout_declare_tensor(3, "r", "t_bias", DTYPE, "texture2d")}
H A Dcopy_channel_offset.glsl17 ${layout_declare_tensor(B, "w", "t_out", DTYPE, STORAGE)}
18 ${layout_declare_tensor(B, "r", "existing_out", DTYPE, STORAGE)}
19 ${layout_declare_tensor(B, "r", "t_in", DTYPE, STORAGE)}
H A Daddmm_optimized.glsl22 ${layout_declare_tensor(B, "w", "out_tensor", DTYPE, "texture3d")}
23 ${layout_declare_tensor(B, "r", "mat1_tensor", DTYPE, "texture3d")}
24 ${layout_declare_tensor(B, "r", "mat2_tensor", DTYPE, "texture3d")}
26 ${layout_declare_tensor(B, "r", "bias_tensor", DTYPE, "texture3d")}
H A Dbuffer_to_buffer.glsl11 ${layout_declare_tensor(0, "w", "out_buf", DTYPE, STORAGE)}
12 ${layout_declare_tensor(1, "r", "in_buf", DTYPE, STORAGE)}
H A Dbuffer_to_nchw.glsl13 ${layout_declare_tensor(0, "w", "nchw_buf", DTYPE, STORAGE)}
14 ${layout_declare_tensor(1, "r", "t_in", DTYPE, STORAGE)}
H A Dnchw_to_buffer.glsl13 ${layout_declare_tensor(0, "w", "t_out", DTYPE, STORAGE)}
14 ${layout_declare_tensor(1, "r", "nchw_in", DTYPE, STORAGE)}
H A Dupsample_nearest2d.glsl20 ${layout_declare_tensor(0, "w", "t_out", DTYPE, STORAGE)}
21 ${layout_declare_tensor(1, "r", "t_in", DTYPE, STORAGE)}
H A Dcopy_offset.glsl17 ${layout_declare_tensor(B, "w", "t_out", DTYPE, STORAGE)}
18 ${layout_declare_tensor(B, "r", "t_in", DTYPE, STORAGE)}
/aosp_15_r20/external/executorch/backends/vulkan/test/glsl/
H A Dreference_matmul.glsl15 ${layout_declare_tensor(0, "w", "t_out", "float", "buffer")}
16 ${layout_declare_tensor(1, "r", "t_mat1", "float", "buffer")}
17 ${layout_declare_tensor(2, "r", "t_mat2", "float", "buffer")}

12