MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / Pow

Function Pow

tensorflow/lite/kernels/internal/reference/reference_ops.h:3367–3375  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

3365
3366template <typename T>
3367inline void Pow(const RuntimeShape& input1_shape, const T* input1_data,
3368 const RuntimeShape& input2_shape, const T* input2_data,
3369 const RuntimeShape& output_shape, T* output_data) {
3370 const int flat_size =
3371 MatchingFlatSize(input1_shape, input2_shape, output_shape);
3372 for (int i = 0; i < flat_size; ++i) {
3373 output_data[i] = std::pow(input1_data[i], input2_data[i]);
3374 }
3375}
3376
3377template <typename T>
3378inline void BroadcastPow4DSlow(const RuntimeShape& unextended_input1_shape,

Callers 1

PowImplFunction · 0.50

Calls 2

MatchingFlatSizeFunction · 0.85
powFunction · 0.50

Tested by

no test coverage detected