| /frameworks/ml/nn/common/operations/ |
| D | Broadcast.cpp | 65 const float* in1, const Shape& shape1, const float* in2, const Shape& shape2,
68 bool binaryOperationFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, in binaryOperationFloat16() argument
71 std::vector<float> in1_float32(getNumberOfElements(shape1)); in binaryOperationFloat16()
77 operationFloat32(in1_float32.data(), shape1, in2_float32.data(), shape2, activation, in binaryOperationFloat16()
84 bool addFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in addFloat32() argument
87 bool needBroadcast = !SameShape(shape1, shape2); in addFloat32()
92 in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), out, \ in addFloat32()
101 in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), out, \ in addFloat32()
111 bool addFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in addFloat16() argument
114 return binaryOperationFloat16(in1, shape1, in2, shape2, activation, out, shapeOut, &addFloat32); in addFloat16()
[all …]
|
| D | MaximumMinimum.cpp | 91 bool eval(const void* in1, const Shape& shape1, const void* in2, const Shape& shape2, in eval() argument
94 switch (shape1.type) { in eval()
96 return evalGeneric(reinterpret_cast<const _Float16*>(in1), shape1, in eval()
101 return evalGeneric(reinterpret_cast<const float*>(in1), shape1, in eval()
106 return evalGeneric(reinterpret_cast<const int32_t*>(in1), shape1, in eval()
111 return evalQuant8(reinterpret_cast<const uint8_t*>(in1), shape1, in eval()
116 LOG(ERROR) << "Unsupported data type: " << toString(shape1.type); in eval()
|
| D | MaximumMinimum.h | 28 bool eval(const void* in1, const Shape& shape1, const void* in2, const Shape& shape2,
|
| /frameworks/ml/nn/common/ |
| D | UtilsTest.cpp | 30 Shape shape1; in TEST() local
32 shape1.dimensions = {4, 3, 2, 1}; in TEST()
39 EXPECT_TRUE(calculateBroadcastedShape(shape1, shape2, &actualOutputShape)); in TEST()
42 EXPECT_TRUE(calculateBroadcastedShape(shape2, shape1, &actualOutputShape)); in TEST()
47 Shape shape1; in TEST() local
49 shape1.dimensions = {5}; in TEST()
53 EXPECT_FALSE(calculateBroadcastedShape(shape1, shape2, &actualOutputShape)); in TEST()
54 EXPECT_FALSE(calculateBroadcastedShape(shape2, shape1, &actualOutputShape)); in TEST()
|
| /frameworks/layoutlib/bridge/src/android/graphics/ |
| D | Region_Delegate.java | 79 public static Area combineShapes(Shape shape1, Shape shape2, int regionOp) { in combineShapes() argument
82 if (shape1 == null) { in combineShapes()
87 Area result = new Area(shape1); in combineShapes()
93 if (shape1 == null) { in combineShapes()
98 Area result = new Area(shape1); in combineShapes()
104 if (shape1 == null) { in combineShapes()
109 Area result = new Area(shape1); in combineShapes()
115 if (shape1 == null) { in combineShapes()
120 Area result = new Area(shape1); in combineShapes()
128 if (shape1 != null) { in combineShapes()
[all …]
|
| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | transpose_conv2d.model.cpp | 10607 auto shape1 = model->addOperand(&type4); in CreateModel_nhwc() local
10620 model->setOperandValue(shape1, shape1_init, sizeof(int32_t) * 4); in CreateModel_nhwc()
10631 …model->addOperation(ANEURALNETWORKS_TRANSPOSE_CONV_2D, {op11, op21, op31, shape1, param3, param4, … in CreateModel_nhwc()
10656 auto shape1 = model->addOperand(&type4); in CreateModel_nhwc_weight_as_input() local
10665 model->setOperandValue(shape1, shape1_init, sizeof(int32_t) * 4); in CreateModel_nhwc_weight_as_input()
10676 …model->addOperation(ANEURALNETWORKS_TRANSPOSE_CONV_2D, {op11, op21, op31, shape1, param3, param4, … in CreateModel_nhwc_weight_as_input()
10701 auto shape1 = model->addOperand(&type4); in CreateModel_nhwc_relaxed() local
10714 model->setOperandValue(shape1, shape1_init, sizeof(int32_t) * 4); in CreateModel_nhwc_relaxed()
10725 …model->addOperation(ANEURALNETWORKS_TRANSPOSE_CONV_2D, {op11, op21, op31, shape1, param3, param4, … in CreateModel_nhwc_relaxed()
10752 auto shape1 = model->addOperand(&type4); in CreateModel_nhwc_relaxed_weight_as_input() local
[all …]
|