Lines Matching refs:tensor_name
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
381 …PetscCallSmartRedis(put_tensor(smartsim->client, tensor_name, strlen(tensor_name), step_array, dim… in TSMonitor_SGS_DD_Training()