mul.h

Go to the documentation of this file.

#ifndef __NN_MUL_H
#define __NN_MUL_H
 
#include <assert.h>
#include <float.h>
 
#include "tensor.h"
#include "ops/mul.h"
#include "ops/mul1.h"
 
 
void NN_mul(Tensor *out, const Tensor *a, const Tensor *b);
 
void NN_mul1(Tensor *out, const Tensor *in, float scalar);
 
void NN_mul_inplace(Tensor *b, const Tensor *a);
 
void NN_mul1_inplace(Tensor *b, float scalar);
 
 
#endif // __NN_MUL_H