Home
last modified time | relevance | path

Searched refs:create_params_buffer (Results 1 – 25 of 26) sorted by relevance

12

/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DConvolution.cpp216 graph.create_params_buffer( in prepack_weights()
385 graph.create_params_buffer(kernel_params), in add_conv2d_node()
386 graph.create_params_buffer(extra_params), in add_conv2d_node()
387 graph.create_params_buffer(out_params), in add_conv2d_node()
482 graph.create_params_buffer(kernel_params), in add_conv1d_node()
483 graph.create_params_buffer(out_params), in add_conv1d_node()
H A DPool.cpp107 graph.create_params_buffer(kernel_params), in add_max_pool2d_node()
179 graph.create_params_buffer(kernel_params), in add_avg_pool2d_node()
180 graph.create_params_buffer(divisor_params)}, in add_avg_pool2d_node()
H A DPad.cpp92 graph.create_params_buffer(pad_param), in add_constant_pad_nd_node()
93 graph.create_params_buffer(fill_value_val)}, in add_constant_pad_nd_node()
H A DArange.cpp100 graph.create_params_buffer(start_val), in add_arange_node()
101 graph.create_params_buffer(step_val)}, in add_arange_node()
H A DBatchNorm.cpp97 graph.create_params_buffer(epsilon), in add_native_batch_norm_node()
98 graph.create_params_buffer(num_texel_per_batch)})); in add_native_batch_norm_node()
H A DUpsample.cpp107 graph.create_params_buffer(input_size), in add_upsample_nearest2d_node()
108 graph.create_params_buffer(rev_scales)}, in add_upsample_nearest2d_node()
H A DBinaryOp.cpp89 graph.create_params_buffer(broadcast_params), in add_binary_op_node()
90 graph.create_params_buffer(alpha_val)}, in add_binary_op_node()
H A DUnaryOp.cpp52 {graph.create_params_buffer(min), graph.create_params_buffer(max)}); in add_unary_op_node()
H A DCopy.cpp60 graph.create_params_buffer(offset_params), in add_copy_offset_node()
171 graph.create_params_buffer(channel_offset_params), in add_copy_channel_offset_node()
H A DSlice.cpp102 graph.create_params_buffer(params)})); in add_slice_tensor_copy_node()
150 {t_out->sizes_ubo(), graph.create_params_buffer(params)})); in add_slice_tensor_copy_node()
H A DLinear.cpp125 graph.create_params_buffer(params), in add_addmm_naive_node()
217 graph.create_params_buffer(params), in add_addmm_optimized_node()
H A DSDPA.cpp95 graph.create_params_buffer(scale_val)}; in add_attn_weight_scale_and_mask_node()
102 graph.create_params_buffer(scale_val)}; in add_attn_weight_scale_and_mask_node()
H A DGridPriors.cpp64 graph.create_params_buffer(param), in add_grid_priors_node()
H A DFull.cpp55 {t_out->sizes_ubo(), graph.create_params_buffer(fill_value_val)}, in add_full_node()
H A DFlip.cpp75 graph.create_params_buffer(dim_bitmap), in add_flip_node()
H A DPermute.cpp95 graph.create_params_buffer(params)}, in add_permute_node()
H A DSelect.cpp120 graph.create_params_buffer( in add_select_int_node()
H A DNativeLayerNorm.cpp112 graph.create_params_buffer(epsilon), in add_native_layer_norm_node()
H A DIndexSelect.cpp100 {t_out->sizes_ubo(), graph.create_params_buffer(params)})); in add_index_select_node()
H A DRepeat.cpp114 {graph.create_params_buffer(repeat_channel_args)}, in add_repeat_channel_node()
/aosp_15_r20/external/executorch/backends/vulkan/runtime/vk_api/memory/
H A DAllocator.h84 VulkanBuffer create_params_buffer(const Block& block);
98 inline VulkanBuffer Allocator::create_params_buffer(const Block& block) { in create_params_buffer() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/api/
H A DResource.h491 VulkanBuffer create_params_buffer(const Block& block);
580 inline VulkanBuffer MemoryAllocator::create_params_buffer(const Block& block) { in create_params_buffer() function
H A DContext.h219 context_p_->adapter_ptr()->vma().create_params_buffer(block)) {} in UniformParamsBuffer()
/aosp_15_r20/external/executorch/backends/vulkan/runtime/api/containers/
H A DParamsBuffer.h34 context_p_->adapter_ptr()->vma().create_params_buffer(block)) {} in ParamsBuffer()
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/
H A DComputeGraph.h549 vkapi::BufferBindInfo create_params_buffer(const Block& data) {

12