Searched refs:calculate_linear_index (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/executorch/kernels/portable/cpu/ |
H A D | op_convolution.cpp | 119 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 D | op_convolution_backward.cpp | 172 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 D | kernel_ops_util.h | 238 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 D | tensor_util.h | 1320 inline size_t calculate_linear_index( in calculate_linear_index() function 1338 using ::executorch::runtime::calculate_linear_index;
|