| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | RotaryEmbedding.cpp | 69 {graph.logical_limits_ubo(xq_out), graph.logical_limits_ubo(xk_out)}, in add_rotary_embedding_node()
|
| H A D | QuantizedLinear.cpp | 108 {graph.logical_limits_ubo(out_W_packed), in add_q_8w_linear_node()
196 {graph.logical_limits_ubo(out_W_packed), in add_q_8w_linear_optimized_node()
319 ubos.append(graph.logical_limits_ubo(out)); in add_q_4w_linear_node()
|
| H A D | Pool.cpp | 105 t_out->logical_limits_ubo(), in add_max_pool2d_node()
177 {t_out->logical_limits_ubo(), in add_avg_pool2d_node()
|
| H A D | SDPA.cpp | 48 graph.logical_limits_ubo(projected), in add_kv_cache_update_node()
100 graph.logical_limits_ubo(attn_weight), in add_attn_weight_scale_and_mask_node()
|
| H A D | RepeatInterleave.cpp | 66 {graph.logical_limits_ubo(in)}, in add_repeat_interleave_node()
|
| H A D | Flip.cpp | 73 graph.logical_limits_ubo(out), in add_flip_node()
|
| H A D | Convolution.cpp | 383 t_out->logical_limits_ubo(), in add_conv2d_node()
480 t_out->logical_limits_ubo(), in add_conv1d_node()
|
| H A D | Permute.cpp | 93 {t_out->logical_limits_ubo(), in add_permute_node()
|
| H A D | Select.cpp | 115 {t_out->logical_limits_ubo(), in add_select_int_node()
|
| H A D | Softmax.cpp | 100 {graph.logical_limits_ubo(out), graph.sizes_ubo(in)}, in add_softmax_node()
|
| H A D | BatchNorm.cpp | 96 {t_out->logical_limits_ubo(), in add_native_batch_norm_node()
|
| H A D | Upsample.cpp | 106 {t_out->logical_limits_ubo(), in add_upsample_nearest2d_node()
|
| H A D | Reduce.cpp | 95 {graph.logical_limits_ubo(in), graph.sizes_ubo(in)}, in add_reduce_node()
|
| H A D | Clone.cpp | 52 {t_out->logical_limits_ubo()}, in add_clone_node()
|
| H A D | NativeLayerNorm.cpp | 110 t_out->logical_limits_ubo(), in add_native_layer_norm_node()
|
| H A D | UnaryOp.cpp | 49 ubos.append({graph.logical_limits_ubo(out)}); in add_unary_op_node()
|
| H A D | MatMul.cpp | 142 graph.logical_limits_ubo(out), in add_matmul_naive_texture3d_node()
|
| H A D | Linear.cpp | 121 graph.logical_limits_ubo(out), in add_addmm_naive_node()
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ |
| H A D | ComputeGraph.h | 349 inline vkapi::BufferBindInfo logical_limits_ubo(const ValueRef idx) {
350 return values_.at(idx).toTensor().logical_limits_ubo();
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/api/containers/ |
| H A D | Tensor.h | 482 const vkapi::BufferBindInfo logical_limits_ubo();
|
| H A D | Tensor.cpp | 630 const vkapi::BufferBindInfo vTensor::logical_limits_ubo() { in logical_limits_ubo() function in vkcompute::api::vTensor
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/utils/ |
| H A D | test_utils.cpp | 347 out.logical_limits_ubo(), in record_matmul_texture3d()
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/ |
| H A D | vulkan_compute_api_test.cpp | 1463 {graph.logical_limits_ubo(a.value), in TEST()
|