ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
ProcessBroadcastShapes.h File Reference

Go to the source code of this file.

Data Structures

struct  onert_micro::execute::pal::NdArrayDesc< N >
 

Namespaces

namespace  onert_micro
 
namespace  onert_micro::execute
 
namespace  onert_micro::execute::pal
 

Functions

template<int N>
void onert_micro::execute::pal::copyDimsToDesc (const core::OMRuntimeShape &input_shape, NdArrayDesc< N > *desc_out)
 
template<int N, int DIM, typename Calc >
std::enable_if< DIM==N-1, void >::type onert_micro::execute::pal::NDOpsHelperImpl (const NdArrayDesc< N > &output, const Calc &calc, int indexes[N])
 
template<int N, int DIM, typename Calc >
std::enable_if< DIM!=N-1, void >::type onert_micro::execute::pal::NDOpsHelperImpl (const NdArrayDesc< N > &output, const Calc &calc, int indexes[N])
 
template<int N, typename Calc >
void onert_micro::execute::pal::NDOpsHelper (const NdArrayDesc< N > &output, const Calc &calc)
 
template<int N>
void onert_micro::execute::pal::NdArrayDescsForElementwiseBroadcast (const core::OMRuntimeShape &input0_shape, const core::OMRuntimeShape &input1_shape, NdArrayDesc< N > *desc0_out, NdArrayDesc< N > *desc1_out)
 
int onert_micro::execute::pal::subscriptToIndex (const NdArrayDesc< 4 > &desc, int i0, int i1, int i2, int i3)
 
int onert_micro::execute::pal::subscriptToIndex (const NdArrayDesc< 5 > &desc, int indexes[5])
 
bool onert_micro::execute::pal::processBroadcastShapes (const core::OMRuntimeShape &shape0, const core::OMRuntimeShape &shape1, core::BinaryArithmeticBroadcastParams *params)