Home
last modified time | relevance | path

Searched refs:num_neurons (Results 1 – 2 of 2) sorted by relevance

/honee/qfunctions/
H A Dsgs_dd_model.h20 CeedInt num_neurons; member
38 const CeedInt num_neurons = sgsdd_ctx->num_neurons; in DataDrivenInference() local
48 CopyN(bias1, V, num_neurons); in DataDrivenInference()
49 MatVecNM(weight1, inputs, num_neurons, num_inputs, CEED_NOTRANSPOSE, V); in DataDrivenInference()
50 LeakyReLU(V, alpha, num_neurons); in DataDrivenInference()
52 MatVecNM(weight2, V, num_outputs, num_neurons, CEED_NOTRANSPOSE, outputs); in DataDrivenInference()
/honee/problems/
H A Dsgs_dd_model.c596 …x)->num_inputs, num_outputs = (*psgsdd_ctx)->num_outputs, num_neurons = (*psgsdd_ctx)->num_neurons; in SgsDDContextFill() local
606 sgsdd_temp->offsets.bias2 = sgsdd_temp->offsets.bias1 + num_neurons; in SgsDDContextFill()
607 sgsdd_temp->offsets.weight1 = sgsdd_temp->offsets.bias2 + num_neurons; in SgsDDContextFill()
608 sgsdd_temp->offsets.weight2 = sgsdd_temp->offsets.weight1 + num_neurons * num_inputs; in SgsDDContextFill()
609 sgsdd_temp->offsets.out_scaling = sgsdd_temp->offsets.weight2 + num_inputs * num_neurons; in SgsDDContextFill()
625 PetscCall(PetscMalloc1(num_inputs * num_neurons, &temp)); in SgsDDContextFill()
628 …Call(TransposeMatrix(temp, &sgsdd_ctx->data[sgsdd_ctx->offsets.weight1], num_inputs, num_neurons)); in SgsDDContextFill()
632 PetscCall(PetscMalloc1(num_outputs * num_neurons, &temp)); in SgsDDContextFill()
635 …PetscCall(TransposeMatrix(temp, &sgsdd_ctx->data[sgsdd_ctx->offsets.weight2], num_neurons, num_out… in SgsDDContextFill()
691 .num_neurons = 20, in SgsDDSetup()