Home
last modified time | relevance | path

Searched refs:linearize_access_indexes (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/executorch/kernels/portable/cpu/util/
H A Dbroadcast_util.h249 size_t linearize_access_indexes(
264 size_t linearize_access_indexes(
301 a_linear_index = linearize_access_indexes(out_indexes, out.dim(), a); in apply_binary_elementwise_fn()
304 b_linear_index = linearize_access_indexes(out_indexes, out.dim(), b); in apply_binary_elementwise_fn()
350 a_linear_index = linearize_access_indexes(out_indexes, out.dim(), a); in apply_ternary_elementwise_fn()
353 b_linear_index = linearize_access_indexes(out_indexes, out.dim(), b); in apply_ternary_elementwise_fn()
356 c_linear_index = linearize_access_indexes(out_indexes, out.dim(), c); in apply_ternary_elementwise_fn()
H A Delementwise_util.h132 a_linear_index = linearize_access_indexes(out_indexes, out.dim(), a); in apply_bitensor_elementwise_fn()
135 b_linear_index = linearize_access_indexes(out_indexes, out.dim(), b); in apply_bitensor_elementwise_fn()
223 a_linear_index = linearize_access_indexes(out_indexes, out.dim(), a); in apply_tritensor_elementwise_fn()
226 b_linear_index = linearize_access_indexes(out_indexes, out.dim(), b); in apply_tritensor_elementwise_fn()
229 c_linear_index = linearize_access_indexes(out_indexes, out.dim(), c); in apply_tritensor_elementwise_fn()
H A Dbroadcast_util.cpp283 size_t linearize_access_indexes( in linearize_access_indexes() function
309 size_t linearize_access_indexes( in linearize_access_indexes() function
313 return linearize_access_indexes( in linearize_access_indexes()
H A Dadvanced_index_util.cpp123 size_t flat_ix = linearize_access_indexes( in query_integral_index()
/aosp_15_r20/external/executorch/kernels/portable/cpu/util/test/
H A Dbroadcast_test.cpp28 using torch::executor::linearize_access_indexes;
177 size_t linear_index = linearize_access_indexes( in TEST()
187 size_t linear_index = linearize_access_indexes( in TEST()
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_masked_select.cpp124 linearize_access_indexes(broadcast_indexes, broadcast_ndim, in); in masked_select_out()
128 linearize_access_indexes(broadcast_indexes, broadcast_ndim, mask); in masked_select_out()
H A Dop_cdist_forward.cpp78 x1_base_ix = linearize_access_indexes(out_base_coord, out.dim(), x1); in cdist()
81 x2_base_ix = linearize_access_indexes(out_base_coord, out.dim(), x2); in cdist()
H A Dop_index_put.cpp144 size_t val_ix = linearize_access_indexes(x_coord, x_dim, values); in index_put_out()
H A Dop_split_with_sizes_copy.cpp126 size_t in_linear_index = linearize_access_indexes( in split_with_sizes_copy_out()