Home
last modified time | relevance | path

Searched refs:shape2 (Results 1 – 4 of 4) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DBroadcast.cpp77 const float* in1, const Shape& shape1, const float* in2, const Shape& shape2,
81 const Shape& shape2, int32_t activation, _Float16* out, in binaryOperationFloat16() argument
85 std::vector<float> in2_float32(getNumberOfElements(shape2)); in binaryOperationFloat16()
89 operationFloat32(in1_float32.data(), shape1, in2_float32.data(), shape2, activation, in binaryOperationFloat16()
96 bool addFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in addFloat32() argument
99 bool needBroadcast = !SameShape(shape1, shape2); in addFloat32()
104 in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), out, \ in addFloat32()
113 in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), out, \ in addFloat32()
123 bool addFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in addFloat16() argument
126 return binaryOperationFloat16(in1, shape1, in2, shape2, activation, out, shapeOut, &addFloat32); in addFloat16()
[all …]
DMaximumMinimum.cpp95 bool eval(const void* in1, const Shape& shape1, const void* in2, const Shape& shape2, in eval() argument
101 reinterpret_cast<const _Float16*>(in2), shape2, isMinimum, in eval()
106 reinterpret_cast<const float*>(in2), shape2, isMinimum, in eval()
111 reinterpret_cast<const int32_t*>(in2), shape2, isMinimum, in eval()
116 reinterpret_cast<const uint8_t*>(in2), shape2, isMinimum, in eval()
121 reinterpret_cast<const int8_t*>(in2), shape2, isMinimum, in eval()
/packages/modules/NeuralNetworks/common/
DUtilsTest.cpp43 Shape shape2; in TEST() local
45 shape2.dimensions = {3, 1, 5}; in TEST()
51 EXPECT_TRUE(calculateBroadcastedShape(shape1, shape2, &actualOutputShape)); in TEST()
54 EXPECT_TRUE(calculateBroadcastedShape(shape2, shape1, &actualOutputShape)); in TEST()
60 Shape shape2; in TEST() local
62 shape2.dimensions = {3}; in TEST()
65 EXPECT_FALSE(calculateBroadcastedShape(shape1, shape2, &actualOutputShape)); in TEST()
66 EXPECT_FALSE(calculateBroadcastedShape(shape2, shape1, &actualOutputShape)); in TEST()
/packages/modules/NeuralNetworks/common/types/operations/include/
DMaximumMinimum.h28 bool eval(const void* in1, const Shape& shape1, const void* in2, const Shape& shape2,