Searched refs:shape1 (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Broadcast.cpp | 77 const float* in1, const Shape& shape1, const float* in2, const Shape& shape2,
80 bool binaryOperationFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, in binaryOperationFloat16() argument
83 std::vector<float> in1_float32(getNumberOfElements(shape1)); 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 …]
|
| D | MaximumMinimum.cpp | 95 bool eval(const void* in1, const Shape& shape1, const void* in2, const Shape& shape2, in eval() argument
98 switch (shape1.type) { in eval()
100 return evalGeneric(reinterpret_cast<const _Float16*>(in1), shape1, in eval()
105 return evalGeneric(reinterpret_cast<const float*>(in1), shape1, in eval()
110 return evalGeneric(reinterpret_cast<const int32_t*>(in1), shape1, in eval()
115 return evalQuant8(reinterpret_cast<const uint8_t*>(in1), shape1, in eval()
120 return evalQuant8(reinterpret_cast<const int8_t*>(in1), shape1, in eval()
125 LOG(ERROR) << "Unsupported data type: " << shape1.type; in eval()
|
| /packages/modules/NeuralNetworks/common/ |
| D | UtilsTest.cpp | 42 Shape shape1; in TEST() local
44 shape1.dimensions = {4, 3, 2, 1}; in TEST()
51 EXPECT_TRUE(calculateBroadcastedShape(shape1, shape2, &actualOutputShape)); in TEST()
54 EXPECT_TRUE(calculateBroadcastedShape(shape2, shape1, &actualOutputShape)); in TEST()
59 Shape shape1; in TEST() local
61 shape1.dimensions = {5}; 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/ |
| D | MaximumMinimum.h | 28 bool eval(const void* in1, const Shape& shape1, const void* in2, const Shape& shape2,
|