Home
last modified time | relevance | path

Searched refs:calculate_linear_index (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_convolution.cpp119 calculate_linear_index(in_coord, in_strides.data(), 4); in conv2d_impl()
123 calculate_linear_index(w_coord, w_strides.data(), 4); in conv2d_impl()
137 calculate_linear_index(out_coord, out_strides.data(), 4); in conv2d_impl()
155 calculate_linear_index(in_coord, in_strides.data(), 4); in conv2d_impl()
174 calculate_linear_index(w_coord, w_strides.data(), 4); in conv2d_impl()
178 calculate_linear_index(out_coord, out_strides.data(), 4); in conv2d_impl()
H A Dop_convolution_backward.cpp172 int64_t out_idx = calculate_linear_index( in conv2d_backward_impl()
199 int64_t in_idx = calculate_linear_index( in conv2d_backward_impl()
202 int64_t weight_idx = calculate_linear_index( in conv2d_backward_impl()
/aosp_15_r20/external/executorch/kernels/portable/cpu/util/
H A Dkernel_ops_util.h238 calculate_linear_index(in_coord, in_strides.data(), in_dim); in kernel_reduction_then_map_2d()
264 calculate_linear_index(out_coord, out_strides.data(), out_dim); in kernel_reduction_then_map_2d()
/aosp_15_r20/external/executorch/runtime/core/exec_aten/util/
H A Dtensor_util.h1320 inline size_t calculate_linear_index( in calculate_linear_index() function
1338 using ::executorch::runtime::calculate_linear_index;