Searched refs:weight (Results 1 – 5 of 5) sorted by relevance
240 const CeedScalar weight = 1 / sqrt(2.); in KMUnpack() local244 A[2][1] = A[1][2] = weight * v[3]; in KMUnpack()245 A[2][0] = A[0][2] = weight * v[4]; in KMUnpack()246 A[1][0] = A[0][1] = weight * v[5]; in KMUnpack()251 const CeedScalar weight = sqrt(2.); in KMPack() local255 v[3] = A[2][1] * weight; in KMPack()256 v[4] = A[2][0] * weight; in KMPack()257 v[5] = A[1][0] * weight; in KMPack()
524 const CeedScalar weight = 1 / sqrt(2.); // Really sqrt(2.) / 2 in KMStrainRate() local528 strain_rate[3] = weight * (grad_velocity[1][2] + grad_velocity[2][1]); in KMStrainRate()529 strain_rate[4] = weight * (grad_velocity[0][2] + grad_velocity[2][0]); in KMStrainRate()530 strain_rate[5] = weight * (grad_velocity[0][1] + grad_velocity[1][0]); in KMStrainRate()
47 m, n = model.net[layer].weight.shape50 model.net[layer].weight[...] = torch.from_numpy(weights[i])[...]
4 font-weight: normal;
441 Note that the weight coefficients are assumed to be in column-major order.