Searched refs:cnum_dims (Results 1 – 2 of 2) sorted by relevance
305 std::unique_ptr<int[]> cnum_dims; in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList() local311 cnum_dims.reset(new int[num_dims_length]); in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList()320 cnum_dims[i] = static_cast<int>(num_dims_elems[i]); in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList()322 if (cnum_dims[i] > 0) { in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList()323 cshapes_ptr += cnum_dims[i]; in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList()330 TFE_OpSetAttrShapeList(op, cname, cdims.get(), cnum_dims.get(), in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList()
279 std::unique_ptr<int[]> cnum_dims; in Java_org_tensorflow_GraphOperationBuilder_setAttrShapeList() local285 cnum_dims.reset(new int[num_dims_length]); in Java_org_tensorflow_GraphOperationBuilder_setAttrShapeList()294 cnum_dims[i] = static_cast<int>(num_dims_elems[i]); in Java_org_tensorflow_GraphOperationBuilder_setAttrShapeList()296 if (cnum_dims[i] > 0) { in Java_org_tensorflow_GraphOperationBuilder_setAttrShapeList()297 cshapes_ptr += cnum_dims[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()