Home
last modified time | relevance | path

Searched refs:shuffled_input_workspace_ptr (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dfully_connected.h197 uint8_t* shuffled_input_workspace_ptr = shuffled_input_workspace_data; in ShuffledFullyConnected() local
212 *shuffled_input_workspace_ptr++ = dst_val; in ShuffledFullyConnected()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dlegacy_optimized_ops.h2267 uint8* shuffled_input_workspace_ptr = shuffled_input_workspace_data; in ShuffledFullyConnected() local
2281 vst1q_u8(shuffled_input_workspace_ptr + 0, val0); in ShuffledFullyConnected()
2282 vst1q_u8(shuffled_input_workspace_ptr + 16, val1); in ShuffledFullyConnected()
2283 vst1q_u8(shuffled_input_workspace_ptr + 32, val2); in ShuffledFullyConnected()
2284 vst1q_u8(shuffled_input_workspace_ptr + 48, val3); in ShuffledFullyConnected()
2285 shuffled_input_workspace_ptr += 64; in ShuffledFullyConnected()
2297 *shuffled_input_workspace_ptr++ = dst_val; in ShuffledFullyConnected()
H A Doptimized_ops.h799 uint8* shuffled_input_workspace_ptr = shuffled_input_workspace_data; in ShuffledFullyConnected() local
813 vst1q_u8(shuffled_input_workspace_ptr + 0, val0); in ShuffledFullyConnected()
814 vst1q_u8(shuffled_input_workspace_ptr + 16, val1); in ShuffledFullyConnected()
815 vst1q_u8(shuffled_input_workspace_ptr + 32, val2); in ShuffledFullyConnected()
816 vst1q_u8(shuffled_input_workspace_ptr + 48, val3); in ShuffledFullyConnected()
817 shuffled_input_workspace_ptr += 64; in ShuffledFullyConnected()
829 *shuffled_input_workspace_ptr++ = dst_val; in ShuffledFullyConnected()