Searched refs:shapeOut (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Broadcast.cpp | 78 int32_t activation, float* out, const Shape& shapeOut)>;
82 const Shape& shapeOut, binaryFunctionFloat32 operationFloat32) { in binaryOperationFloat16() argument
87 std::vector<float> out_float32(getNumberOfElements(shapeOut)); in binaryOperationFloat16()
90 out_float32.data(), shapeOut); in binaryOperationFloat16()
97 int32_t activation, float* out, const Shape& shapeOut) { in addFloat32() argument
105 convertShapeToDims(shapeOut)) in addFloat32()
114 convertShapeToDims(shapeOut)) in addFloat32()
124 int32_t activation, _Float16* out, const Shape& shapeOut) { in addFloat16() argument
126 return binaryOperationFloat16(in1, shape1, in2, shape2, activation, out, shapeOut, &addFloat32); in addFloat16()
131 int32_t activation, T* out, const Shape& shapeOut) { in addQuant8() argument
[all …]
|