This patch corrects the gtests for unpack op.
Signed-off-by: b.devansh <b.devansh@samsung.com>
model->setOperandValue(axis, axis_init, sizeof(int32_t) * 1);
static int32_t num_splits_init[] = {3};
model->setOperandValue(num_splits, num_splits_init, sizeof(int32_t) * 1);
- model->addOperationEx(ANEURALNETWORKS_UNPACK_EX, {input, axis, num_splits}, {output1, output2, output3});
+ model->addOperationEx(ANEURALNETWORKS_UNPACK_EX, {input, num_splits, axis}, {output1, output2, output3});
// Phase 3, inputs and outputs
model->identifyInputsAndOutputs(
{input},
model->setOperandValue(axis, axis_init, sizeof(int32_t) * 1);
static int32_t num_splits_init[] = {3};
model->setOperandValue(num_splits, num_splits_init, sizeof(int32_t) * 1);
- model->addOperationEx(ANEURALNETWORKS_UNPACK_EX, {input, axis, num_splits}, {output1, output2, output3});
+ model->addOperationEx(ANEURALNETWORKS_UNPACK_EX, {input, num_splits, axis}, {output1, output2, output3});
// Phase 3, inputs and outputs
model->identifyInputsAndOutputs(
{input},
out1 = Output("output1", "TENSOR_FLOAT32", "{2, 3, 4}")
out2 = Output("output2", "TENSOR_FLOAT32", "{2, 3, 4}")
out3 = Output("output3", "TENSOR_FLOAT32", "{2, 3, 4}")
-model = model.Operation("UNPACK_EX", input, axis, num_splits).To([out1, out2, out3])
+model = model.Operation("UNPACK_EX", input, num_splits, axis).To([out1, out2, out3])
input0 = {input: # input 0
[ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17,
out1 = Output("output1", "TENSOR_FLOAT32", "{3, 2, 4}")
out2 = Output("output2", "TENSOR_FLOAT32", "{3, 2, 4}")
out3 = Output("output3", "TENSOR_FLOAT32", "{3, 2, 4}")
-model = model.Operation("UNPACK_EX", input, axis, num_splits).To([out1, out2, out3])
+model = model.Operation("UNPACK_EX", input, num_splits, axis).To([out1, out2, out3])
input0 = {input: # input 0
[ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17,