ONE - On-device Neural Engine
Loading...
Searching...
No Matches
ProcessBroadcastShapes.h File Reference

Go to the source code of this file.

Data Structures

struct  luci_interpreter_pal::NdArrayDesc< N >
 

Namespaces

namespace  luci_interpreter_pal
 

Functions

template<int N>
void luci_interpreter_pal::copyDimsToDesc (const luci_interpreter::RuntimeShape &input_shape, NdArrayDesc< N > *desc_out)
 
template<int N, int DIM, typename Calc >
std::enable_if< DIM==N-1, void >::type luci_interpreter_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 luci_interpreter_pal::NDOpsHelperImpl (const NdArrayDesc< N > &output, const Calc &calc, int indexes[N])
 
template<int N, typename Calc >
void luci_interpreter_pal::NDOpsHelper (const NdArrayDesc< N > &output, const Calc &calc)
 
template<int N>
void luci_interpreter_pal::NdArrayDescsForElementwiseBroadcast (const luci_interpreter::RuntimeShape &input0_shape, const luci_interpreter::RuntimeShape &input1_shape, NdArrayDesc< N > *desc0_out, NdArrayDesc< N > *desc1_out)
 
int luci_interpreter_pal::subscriptToIndex (const NdArrayDesc< 4 > &desc, int i0, int i1, int i2, int i3)
 
int luci_interpreter_pal::subscriptToIndex (const NdArrayDesc< 5 > &desc, int indexes[5])
 
bool luci_interpreter_pal::ProcessBroadcastShapes (const luci_interpreter::RuntimeShape &shape0, const luci_interpreter::RuntimeShape &shape1, luci_interpreter_pal::ArithmeticParams *params)