Searched refs:weight2 (Results 1 – 2 of 2) sorted by relevance
26 size_t weight1, weight2; member45 const CeedScalar *weight2 = &sgsdd_ctx->data[sgsdd_ctx->offsets.weight2]; in DataDrivenInference() local52 MatVecNM(weight2, V, num_outputs, num_neurons, CEED_NOTRANSPOSE, outputs); in DataDrivenInference()
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()635 …PetscCall(TransposeMatrix(temp, &sgsdd_ctx->data[sgsdd_ctx->offsets.weight2], num_neurons, num_out… in SgsDDContextFill()