• Home
  • Raw
  • Download

Lines Matching refs:Shape

77         const float* in1, const Shape& shape1, const float* in2, const Shape& shape2,
78 int32_t activation, float* out, const Shape& shapeOut)>;
80 bool binaryOperationFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, in binaryOperationFloat16()
81 const Shape& shape2, int32_t activation, _Float16* out, in binaryOperationFloat16()
82 const Shape& shapeOut, binaryFunctionFloat32 operationFloat32) { in binaryOperationFloat16()
96 bool addFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in addFloat32()
97 int32_t activation, float* out, const Shape& shapeOut) { in addFloat32()
123 bool addFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in addFloat16()
124 int32_t activation, _Float16* out, const Shape& shapeOut) { in addFloat16()
130 bool addQuant8(const T* in1, const Shape& shape1, const T* in2, const Shape& shape2, in addQuant8()
131 int32_t activation, T* out, const Shape& shapeOut) { in addQuant8()
211 bool executeInt32(const int32_t* aData, const Shape& aShape, const int32_t* bData, in executeInt32()
212 const Shape& bShape, int32_t activation, int32_t* outputData, in executeInt32()
213 const Shape& outputShape, int32_t func(int32_t, int32_t)) { in executeInt32()
235 bool mulFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in mulFloat32()
236 int32_t activation, float* out, const Shape& shapeOut) { in mulFloat32()
262 bool mulFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in mulFloat16()
263 int32_t activation, _Float16* out, const Shape& shapeOut) { in mulFloat16()
269 bool mulQuant8(const T* in1, const Shape& shape1, const T* in2, const Shape& shape2, in mulQuant8()
270 int32_t activation, T* out, const Shape& shapeOut) { in mulQuant8()
316 bool subFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in subFloat32()
317 int32_t activation, float* out, const Shape& shapeOut) { in subFloat32()
333 bool subFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in subFloat16()
334 int32_t activation, _Float16* out, const Shape& shapeOut) { in subFloat16()
340 bool subQuant8(const T* in1, const Shape& shape1, const T* in2, const Shape& shape2, in subQuant8()
341 int32_t activation, T* out, const Shape& shapeOut) { in subQuant8()
412 bool divFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in divFloat32()
413 int32_t activation, float* out, const Shape& shapeOut) { in divFloat32()
433 bool divFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in divFloat16()
434 int32_t activation, _Float16* out, const Shape& shapeOut) { in divFloat16()
459 Shape output = context->getOutputShape(kOutputTensor); in validate()
460 Shape input1 = context->getInputShape(kInputTensor1); in validate()
461 Shape input2 = context->getInputShape(kInputTensor2); in validate()
473 const Shape& input1 = context->getInputShape(kInputTensor1); in validate()
474 const Shape& input2 = context->getInputShape(kInputTensor2); in validate()
486 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
487 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
488 Shape output = context->getOutputShape(kOutputTensor); in prepare()