Searched refs:cdims (Results 1 – 2 of 2) sorted by relevance
120 std::unique_ptr<int64_t[]> cdims(new int64_t[num_dims]); in Java_org_tensorflow_Operation_shape() local121 TF_GraphGetTensorShape(graph, output, cdims.get(), static_cast<int>(num_dims), in Java_org_tensorflow_Operation_shape()132 dims[i] = static_cast<jlong>(cdims[i]); in Java_org_tensorflow_Operation_shape()
272 std::unique_ptr<int64_t* []> cdims; in Java_org_tensorflow_OperationBuilder_setAttrShapeList() local278 cdims.reset(new int64_t*[num_dims_length]); in Java_org_tensorflow_OperationBuilder_setAttrShapeList()289 cdims[i] = cshapes_ptr; in Java_org_tensorflow_OperationBuilder_setAttrShapeList()297 TF_SetAttrShapeList(d, cname, cdims.get(), cnum_dims.get(), num_dims_length); in Java_org_tensorflow_OperationBuilder_setAttrShapeList()