Lines Matching refs:Shape

65         const float* in1, const Shape& shape1, const float* in2, const Shape& shape2,
66 int32_t activation, float* out, const Shape& shapeOut)>;
68 bool binaryOperationFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, in binaryOperationFloat16()
69 const Shape& shape2, int32_t activation, _Float16* out, in binaryOperationFloat16()
70 const Shape& shapeOut, binaryFunctionFloat32 operationFloat32) { in binaryOperationFloat16()
84 bool addFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in addFloat32()
85 int32_t activation, float* out, const Shape& shapeOut) { in addFloat32()
111 bool addFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in addFloat16()
112 int32_t activation, _Float16* out, const Shape& shapeOut) { in addFloat16()
117 bool addQuant8(const uint8_t* in1, const Shape& shape1, const uint8_t* in2, const Shape& shape2, in addQuant8()
118 int32_t activation, uint8_t* out, const Shape& shapeOut) { in addQuant8()
182 bool mulFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in mulFloat32()
183 int32_t activation, float* out, const Shape& shapeOut) { in mulFloat32()
209 bool mulFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in mulFloat16()
210 int32_t activation, _Float16* out, const Shape& shapeOut) { in mulFloat16()
215 bool mulQuant8(const uint8_t* in1, const Shape& shape1, const uint8_t* in2, const Shape& shape2, in mulQuant8()
216 int32_t activation, uint8_t* out, const Shape& shapeOut) { in mulQuant8()
243 bool subFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in subFloat32()
244 int32_t activation, float* out, const Shape& shapeOut) { in subFloat32()
260 bool subFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in subFloat16()
261 int32_t activation, _Float16* out, const Shape& shapeOut) { in subFloat16()
266 bool subQuant8(const uint8_t* in1, const Shape& shape1, const uint8_t* in2, const Shape& shape2, in subQuant8()
267 int32_t activation, uint8_t* out, const Shape& shapeOut) { in subQuant8()
321 bool divFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in divFloat32()
322 int32_t activation, float* out, const Shape& shapeOut) { in divFloat32()
342 bool divFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in divFloat16()
343 int32_t activation, _Float16* out, const Shape& shapeOut) { in divFloat16()
367 Shape output = context->getOutputShape(kOutputTensor); in validate()
368 Shape input1 = context->getInputShape(kInputTensor1); in validate()
369 Shape input2 = context->getInputShape(kInputTensor2); in validate()
383 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
384 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
385 Shape output = context->getOutputShape(kOutputTensor); in prepare()