Fix some issues with optimized xtensa kernels build.
PiperOrigin-RevId: 318355987 Change-Id: I6122de816e0ae691fad2ff721ff6186cd2c4aca2
This commit is contained in:
parent
e25272e743
commit
f0b9d27774
@ -47,8 +47,8 @@ limitations under the License.
|
||||
#include <xtensa/tie/xt_FP.h>
|
||||
#endif
|
||||
|
||||
#include "tensorflow/lite/micro/kernels/xtensa_hifimini/xa_nnlib/algo/include/xa_nnlib_err_chk.h"
|
||||
#include "tensorflow/lite/micro/kernels/xtensa_hifimini/xa_nnlib/algo/include/xa_nnlib_kernels_api.h"
|
||||
#include "tensorflow/lite/micro/kernels/xtensa_hifimini/xa_nnlib/algo/common/include/xa_nnlib_err_chk.h"
|
||||
#include "tensorflow/lite/micro/kernels/xtensa_hifimini/xa_nnlib/include/nnlib/xa_nnlib_kernels_api.h"
|
||||
#include "tensorflow/lite/micro/kernels/xtensa_hifimini/xa_nnlib/include/nnlib/xa_nnlib_standards.h"
|
||||
#include "tensorflow/lite/micro/kernels/xtensa_hifimini/xa_nnlib/include/xa_type_def.h"
|
||||
|
||||
|
||||
@ -498,7 +498,7 @@ WORD32 xa_nn_vec_softmax_asym8u_8(UWORD8 *__restrict__ pOut,
|
||||
ae_p24x2s p_min = AE_ZEROP48();
|
||||
ae_p24x2s p_max = AE_MOVPA24(255);
|
||||
|
||||
for (i = 0; i<vec_length> > 1; i++) {
|
||||
for (i = 0; i<vec_length >> 1; i++) {
|
||||
int out;
|
||||
|
||||
p_exp = *(ae_p24x2f *)&pExp[2 * i];
|
||||
@ -713,7 +713,7 @@ WORD32 xa_nn_vec_softmax_asym8s_8(WORD8 *__restrict__ pOut,
|
||||
ae_p24x2s p_min = AE_MOVPA24(-128);
|
||||
ae_p24x2s p_max = AE_MOVPA24(127);
|
||||
|
||||
for (i = 0; i<vec_length> > 1; i++) {
|
||||
for (i = 0; i<vec_length >> 1; i++) {
|
||||
int out;
|
||||
|
||||
p_exp = *(ae_p24x2f *)&pExp[2 * i];
|
||||
@ -928,7 +928,7 @@ WORD32 xa_nn_vec_softmax_asym8s_16(WORD16 *__restrict__ pOut,
|
||||
ae_p24x2s p_min = AE_MOVPA24(-32768);
|
||||
ae_p24x2s p_max = AE_MOVPA24(32767);
|
||||
|
||||
for (i = 0; i<vec_length> > 1; i++) {
|
||||
for (i = 0; i<vec_length >> 1; i++) {
|
||||
int out;
|
||||
|
||||
p_exp = *(ae_p24x2f *)&pExp[2 * i];
|
||||
|
||||
@ -47,7 +47,7 @@ typedef char Int8;
|
||||
typedef int16_t Int16;
|
||||
typedef int Int32;
|
||||
typedef int Int24;
|
||||
typedef int64_t int Int64;
|
||||
typedef int64_t Int64;
|
||||
typedef int Bool;
|
||||
typedef float Flt32;
|
||||
|
||||
|
||||
@ -37,6 +37,8 @@ limitations under the License.
|
||||
#ifndef __XA_TYPE_DEF_H__
|
||||
#define __XA_TYPE_DEF_H__
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
/****************************************************************************/
|
||||
/* types type define prefix examples bytes */
|
||||
/************************ *********** ****** **************** ***** */
|
||||
@ -68,7 +70,7 @@ typedef uint64_t* pUWORD40; /* pum pUWORD40 pum_count 5 */
|
||||
typedef int64_t WORD64; /* h WORD64 h_count 8 */
|
||||
typedef int64_t* pWORD64; /* ph pWORD64 ph_count 8 */
|
||||
typedef uint64_t UWORD64; /* uh UWORD64 uh_count 8 */
|
||||
typedef uint64_tg* pUWORD64; /* puh pUWORD64 puh_count 8 */
|
||||
typedef uint64_t* pUWORD64; /* puh pUWORD64 puh_count 8 */
|
||||
|
||||
typedef float FLOAT32; /* f FLOAT32 f_count 4 */
|
||||
typedef float* pFLOAT32; /* pf pFLOAT32 pf_count 4 */
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user