Searched refs:num_dims_length (Results 1 – 2 of 2) sorted by relevance
306 const int num_dims_length = env->GetArrayLength(num_dims); in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList() local307 if (num_dims_length > 0) { in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList()310 cdims.reset(new const int64_t*[num_dims_length]); in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList()311 cnum_dims.reset(new int[num_dims_length]); in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList()319 for (int i = 0; i < num_dims_length; ++i) { in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList()331 num_dims_length, status); in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList()
280 const int num_dims_length = env->GetArrayLength(num_dims); in Java_org_tensorflow_GraphOperationBuilder_setAttrShapeList() local281 if (num_dims_length > 0) { in Java_org_tensorflow_GraphOperationBuilder_setAttrShapeList()284 cdims.reset(new int64_t*[num_dims_length]); in Java_org_tensorflow_GraphOperationBuilder_setAttrShapeList()285 cnum_dims.reset(new int[num_dims_length]); in Java_org_tensorflow_GraphOperationBuilder_setAttrShapeList()293 for (int i = 0; i < num_dims_length; ++i) { in Java_org_tensorflow_GraphOperationBuilder_setAttrShapeList()303 TF_SetAttrShapeList(d, cname, cdims.get(), cnum_dims.get(), num_dims_length); in Java_org_tensorflow_GraphOperationBuilder_setAttrShapeList()