Home
last modified time | relevance | path

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 Dfully_connected.h172 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 Dlegacy_reference_ops.h514 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 Dlegacy_optimized_ops.h2219 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 Doptimized_ops.h434 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 …]