Searched refs:shuffled_input_workspace_data (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
H A D | fully_connected.h | 172 int16_t* output_data, uint8_t* shuffled_input_workspace_data) { in ShuffledFullyConnected() argument 197 uint8_t* shuffled_input_workspace_ptr = shuffled_input_workspace_data; in ShuffledFullyConnected() 200 shuffled_input_workspace_data[i] = input_data[i] ^ 0x80; in ShuffledFullyConnected() 232 reinterpret_cast<const int8_t*>(shuffled_input_workspace_data); in ShuffledFullyConnected() 272 reinterpret_cast<const int8_t*>(shuffled_input_workspace_data); in ShuffledFullyConnected()
|
H A D | legacy_reference_ops.h | 514 int16* output_data, uint8* shuffled_input_workspace_data, in ShuffledFullyConnected() argument 519 shuffled_input_workspace_data); in ShuffledFullyConnected() 528 uint8* shuffled_input_workspace_data, in ShuffledFullyConnected() argument 541 shuffled_input_workspace_data, gemmlowp_context); in ShuffledFullyConnected()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
H A D | legacy_optimized_ops.h | 2219 int16* output_data, uint8* shuffled_input_workspace_data, in ShuffledFullyConnected() argument 2259 vst1q_u8(shuffled_input_workspace_data + i, val); in ShuffledFullyConnected() 2263 shuffled_input_workspace_data[i] = input_data[i] ^ 0x80; in ShuffledFullyConnected() 2267 uint8* shuffled_input_workspace_ptr = shuffled_input_workspace_data; in ShuffledFullyConnected() 2314 shuffled_input_workspace_data, int8_shuffled_weights_data, batches, in ShuffledFullyConnected() 2329 shuffled_input_workspace_data, in ShuffledFullyConnected() 2345 uint8* shuffled_input_workspace_data, in ShuffledFullyConnected() argument 2358 shuffled_input_workspace_data, gemmlowp_context); in ShuffledFullyConnected()
|
H A D | optimized_ops.h | 434 const uint8* shuffled_input_workspace_data, in ShuffledFullyConnectedWorkerImpl() argument 456 vreinterpretq_s8_u8(vld1q_u8(shuffled_input_workspace_data + d)); in ShuffledFullyConnectedWorkerImpl() 512 reinterpret_cast<const int8*>(shuffled_input_workspace_data); in ShuffledFullyConnectedWorkerImpl() 618 reinterpret_cast<const int8*>(shuffled_input_workspace_data); in ShuffledFullyConnectedWorkerImpl() 658 reinterpret_cast<const int8*>(shuffled_input_workspace_data); in ShuffledFullyConnectedWorkerImpl() 752 int16* output_data, uint8* shuffled_input_workspace_data, in ShuffledFullyConnected() argument 791 vst1q_u8(shuffled_input_workspace_data + i, val); in ShuffledFullyConnected() 795 shuffled_input_workspace_data[i] = input_data[i] ^ 0x80; in ShuffledFullyConnected() 799 uint8* shuffled_input_workspace_ptr = shuffled_input_workspace_data; in ShuffledFullyConnected() 847 shuffled_input_workspace_data, int8_shuffled_weights_data, batches, in ShuffledFullyConnected() [all …]
|