Add Conv2Dsame and Conv2Dsame_stride test circuits and Conv2Dsame.circom

This commit is contained in:
drCathieSo.eth
2024-02-04 01:31:22 +08:00
parent 1b49b9bbac
commit 5713443fb1
7 changed files with 943 additions and 0 deletions

37
test/Conv2Dsame.js Normal file
View File

@@ -0,0 +1,37 @@
const chai = require("chai");
const path = require("path");
const wasm_tester = require("circom_tester").wasm;
const F1Field = require("ffjavascript").F1Field;
const Scalar = require("ffjavascript").Scalar;
exports.p = Scalar.fromString("21888242871839275222246405745257275088548364400416034343698204186575808495617");
const Fr = new F1Field(exports.p);
const assert = chai.assert;
describe("Conv2Dsame layer test", function () {
this.timeout(100000000);
it("(5,5,3) -> (5,5,2)", async () => {
const INPUT = require("../models/Conv2Dsame_input.json");
const circuit = await wasm_tester(path.join(__dirname, "circuits", "Conv2Dsame_test.circom"));
const witness = await circuit.calculateWitness(INPUT, true);
assert(Fr.eq(Fr.e(witness[0]),Fr.e(1)));
});
it("(10,10,3) -> (4,4,2)", async () => {
const INPUT = require("../models/Conv2Dsame_stride_input.json");
const circuit = await wasm_tester(path.join(__dirname, "circuits", "Conv2Dsame_stride_test.circom"));
const witness = await circuit.calculateWitness(INPUT, true);
assert(Fr.eq(Fr.e(witness[0]),Fr.e(1)));
});
});

View File

@@ -0,0 +1,5 @@
pragma circom 2.0.0;
include "../../circuits/Conv2Dsame.circom";
component main = Conv2Dsame(10, 10, 3, 2, 4, 3, 10**36);

View File

@@ -0,0 +1,5 @@
pragma circom 2.0.0;
include "../../circuits/Conv2Dsame.circom";
component main = Conv2Dsame(5, 5, 3, 2, 3, 1, 10**36);