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 …]
|