Home
last modified time | relevance | path

Searched defs:packed_dim (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/glsl/
H A Dindexing_utils.h66 int packed_dim = 0; in find_packed_dim() local
81 ivec4 tidx_to_nchwi(const ivec4 tidx, const ivec4 sizes, const int packed_dim) { in tidx_to_nchwi()
99 const int packed_dim) { in tidx_to_4bufi()
122 ivec4 bufi_to_tidx(int bufi, const ivec4 strides, const int packed_dim) { in bufi_to_tidx()
138 int packed_dim = find_packed_dim(strides); in bufi_to_tidx() local
151 const int packed_dim) { in lpos_to_tidx()
170 const int packed_dim) { in tidx_to_lpos()
190 const int packed_dim) { in tidx_to_pos()
213 const int packed_dim) { in tidx_to_posi()
294 bool pos_out_of_bounds(ivec3 pos, ivec4 sizes, int packed_dim) { in pos_out_of_bounds()
[all …]
/aosp_15_r20/external/executorch/backends/vulkan/test/glsl/
H A Dindexing_utils.h46 ivec4 get_texel_nchw_buffer_ixs(ivec4 idx, ivec4 sizes, int packed_dim) { in get_texel_nchw_buffer_ixs()
66 bool pos_out_of_bounds(ivec3 pos, ivec4 sizes, int packed_dim) { in pos_out_of_bounds()
82 ivec4 to_tensor_idx(ivec3 pos, ivec4 sizes, int packed_dim) { in to_tensor_idx()
/aosp_15_r20/external/executorch/backends/vulkan/runtime/api/containers/
H A DTensor.cpp19 const auto packed_dim = utils::to_packed_dim<int32_t>(memory_layout); in calculate_sizes() local
26 const int32_t packed_dim) { in calculate_dim_order()
143 const int32_t packed_dim) { in calculate_padded_sizes()
167 const int32_t packed_dim) { in calculate_image_extents()
280 const int32_t packed_dim, in vTensorStorage()
H A DTensor.h400 inline int32_t packed_dim() const { in packed_dim() function
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/utils/
H A DShaderNameUtils.cpp76 void add_packed_dim_suffix(std::string& kernel_name, const int32_t packed_dim) { in add_packed_dim_suffix()
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/utils/
H A DTensorUtils.cpp48 bool check_packed_dim_is(const api::vTensor& t, const int32_t packed_dim) { in check_packed_dim_is()
/aosp_15_r20/external/executorch/kernels/quantized/cpu/
H A Dembeddingxb.cpp55 int32_t packed_dim, in get_embedding_dim()
/aosp_15_r20/external/executorch/backends/vulkan/test/
H A Dvulkan_compute_api_test.cpp203 const int32_t packed_dim = std::get<1>(test_case); in TEST_F() local
227 const int32_t packed_dim = static_cast<int32_t>(layout); in TEST_F() local