Searched refs:b_linear_index (Results 1 – 2 of 2) sorted by relevance
| /external/executorch/kernels/portable/cpu/util/ |
| D | broadcast_util.h | 294 size_t b_linear_index = i; in apply_binary_elementwise_fn() local
304 b_linear_index = linearize_access_indexes(out_indexes, out.dim(), b); in apply_binary_elementwise_fn()
308 data_out[i] = compute_fun(data_a[a_linear_index], data_b[b_linear_index]); in apply_binary_elementwise_fn()
342 size_t b_linear_index = i; in apply_ternary_elementwise_fn() local
353 b_linear_index = linearize_access_indexes(out_indexes, out.dim(), b); in apply_ternary_elementwise_fn()
361 data_a[a_linear_index], data_b[b_linear_index], data_c[c_linear_index]); in apply_ternary_elementwise_fn()
|
| D | elementwise_util.h | 125 size_t b_linear_index = i; in apply_bitensor_elementwise_fn() local
135 b_linear_index = linearize_access_indexes(out_indexes, out.dim(), b); in apply_bitensor_elementwise_fn()
141 load_b_to_common(&data_b[b_linear_index * b_element_size])); in apply_bitensor_elementwise_fn()
215 size_t b_linear_index = i; in apply_tritensor_elementwise_fn() local
226 b_linear_index = linearize_access_indexes(out_indexes, out.dim(), b); in apply_tritensor_elementwise_fn()
235 load_b_to_common(&data_b[b_linear_index * b_element_size]), in apply_tritensor_elementwise_fn()
|