Searched refs:array_info (Results 1 – 1 of 1) sorted by relevance
238 PetscInt64 array_info[6] = {0}, num_features = 6; in SGS_DD_TrainingSetup() local240 array_info[0] = sgs_dd_train->training_data_array_dims[0]; in SGS_DD_TrainingSetup()241 array_info[1] = sgs_dd_train->training_data_array_dims[1]; in SGS_DD_TrainingSetup()242 array_info[2] = num_features; in SGS_DD_TrainingSetup()243 array_info[3] = num_ranks; in SGS_DD_TrainingSetup()244 array_info[4] = smartsim->collocated_database_num_ranks; in SGS_DD_TrainingSetup()245 array_info[5] = rank; in SGS_DD_TrainingSetup()248 …rtRedis(put_tensor(smartsim->client, tensor_name, strlen(tensor_name), array_info, &array_info_dim… in SGS_DD_TrainingSetup()