ONE - On-device Neural Engine
Loading...
Searching...
No Matches
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)