Searched refs:shuffled_input_workspace_data (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
D | fully_connected.h | 173 int16* output_data, uint8* shuffled_input_workspace_data, in ShuffledFullyConnected() argument 200 uint8* shuffled_input_workspace_ptr = shuffled_input_workspace_data; in ShuffledFullyConnected() 203 shuffled_input_workspace_data[i] = input_data[i] ^ 0x80; in ShuffledFullyConnected() 235 reinterpret_cast<const int8*>(shuffled_input_workspace_data); in ShuffledFullyConnected() 275 reinterpret_cast<const int8*>(shuffled_input_workspace_data); in ShuffledFullyConnected()
|
D | legacy_reference_ops.h | 478 uint8* shuffled_input_workspace_data, gemmlowp::GemmContext* gemm_context) { in ShuffledFullyConnected() argument 490 shuffled_input_workspace_data, gemm_context); in ShuffledFullyConnected()
|
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
D | optimized_ops.h | 1374 const uint8* shuffled_input_workspace_data, in ShuffledFullyConnectedWorkerImpl() argument 1396 vreinterpretq_s8_u8(vld1q_u8(shuffled_input_workspace_data + d)); in ShuffledFullyConnectedWorkerImpl() 1452 reinterpret_cast<const int8*>(shuffled_input_workspace_data); in ShuffledFullyConnectedWorkerImpl() 1558 reinterpret_cast<const int8*>(shuffled_input_workspace_data); in ShuffledFullyConnectedWorkerImpl() 1598 reinterpret_cast<const int8*>(shuffled_input_workspace_data); in ShuffledFullyConnectedWorkerImpl() 1692 int16* output_data, uint8* shuffled_input_workspace_data, in ShuffledFullyConnected() argument 1732 vst1q_u8(shuffled_input_workspace_data + i, val); in ShuffledFullyConnected() 1736 shuffled_input_workspace_data[i] = input_data[i] ^ 0x80; in ShuffledFullyConnected() 1740 uint8* shuffled_input_workspace_ptr = shuffled_input_workspace_data; in ShuffledFullyConnected() 1787 shuffled_input_workspace_data, int8_shuffled_weights_data, batches, in ShuffledFullyConnected() [all …]
|
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
D | optimized_ops.h | 1318 const uint8* shuffled_input_workspace_data, in ShuffledFullyConnectedWorkerImpl() argument 1340 vreinterpretq_s8_u8(vld1q_u8(shuffled_input_workspace_data + d)); in ShuffledFullyConnectedWorkerImpl() 1396 reinterpret_cast<const int8*>(shuffled_input_workspace_data); in ShuffledFullyConnectedWorkerImpl() 1502 reinterpret_cast<const int8*>(shuffled_input_workspace_data); in ShuffledFullyConnectedWorkerImpl() 1542 reinterpret_cast<const int8*>(shuffled_input_workspace_data); in ShuffledFullyConnectedWorkerImpl() 1636 int16* output_data, uint8* shuffled_input_workspace_data, in ShuffledFullyConnected() argument 1676 vst1q_u8(shuffled_input_workspace_data + i, val); in ShuffledFullyConnected() 1680 shuffled_input_workspace_data[i] = input_data[i] ^ 0x80; in ShuffledFullyConnected() 1684 uint8* shuffled_input_workspace_ptr = shuffled_input_workspace_data; in ShuffledFullyConnected() 1731 shuffled_input_workspace_data, int8_shuffled_weights_data, batches, in ShuffledFullyConnected() [all …]
|
D | legacy_optimized_ops.h | 432 uint8* shuffled_input_workspace_data, gemmlowp::GemmContext* gemm_context) { in ShuffledFullyConnected() argument 444 shuffled_input_workspace_data, gemm_context); in ShuffledFullyConnected()
|