Searched refs:tensor_name (Results 1 – 1 of 1) sorted by relevance
| /honee/src/smartsim/ |
| H A D | sgs_dd_training.c | 236 const char tensor_name[] = "sizeInfo"; in SGS_DD_TrainingSetup() local
248 …PetscCallSmartRedis(put_tensor(smartsim->client, tensor_name, strlen(tensor_name), array_info, &ar… in SGS_DD_TrainingSetup()
250 PetscCall(SmartRedisVerifyPutTensor(smartsim->client, tensor_name, strlen(tensor_name))); in SGS_DD_TrainingSetup()
255 const char tensor_name[] = "tensor-ow"; in SGS_DD_TrainingSetup() local
260 …PetscCallSmartRedis(put_tensor(smartsim->client, tensor_name, strlen(tensor_name), tensor_overwrit… in SGS_DD_TrainingSetup()
262 PetscCall(SmartRedisVerifyPutTensor(smartsim->client, tensor_name, strlen(tensor_name))); in SGS_DD_TrainingSetup()
267 const char tensor_name[] = "num_filter_widths"; in SGS_DD_TrainingSetup() local
272 …PetscCallSmartRedis(put_tensor(smartsim->client, tensor_name, strlen(tensor_name), &num_filter_wid… in SGS_DD_TrainingSetup()
274 PetscCall(SmartRedisVerifyPutTensor(smartsim->client, tensor_name, strlen(tensor_name))); in SGS_DD_TrainingSetup()
376 const char tensor_name[] = "step"; in TSMonitor_SGS_DD_Training() local
[all …]
|