Home
last modified time | relevance | path

Searched refs:cdims (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/java/src/main/native/
Dgraph_operation_jni.cc113 std::unique_ptr<int64_t[]> cdims(new int64_t[num_dims]); in Java_org_tensorflow_GraphOperation_shape() local
114 TF_GraphGetTensorShape(graph, output, cdims.get(), static_cast<int>(num_dims), in Java_org_tensorflow_GraphOperation_shape()
125 dims[i] = static_cast<jlong>(cdims[i]); in Java_org_tensorflow_GraphOperation_shape()
Deager_operation_builder_jni.cc304 std::unique_ptr<const int64_t*[]> cdims; in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList() local
310 cdims.reset(new const int64_t*[num_dims_length]); in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList()
321 cdims[i] = cshapes_ptr; in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList()
330 TFE_OpSetAttrShapeList(op, cname, cdims.get(), cnum_dims.get(), in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList()
Dgraph_operation_builder_jni.cc278 std::unique_ptr<int64_t*[]> cdims; in Java_org_tensorflow_GraphOperationBuilder_setAttrShapeList() local
284 cdims.reset(new int64_t*[num_dims_length]); in Java_org_tensorflow_GraphOperationBuilder_setAttrShapeList()
295 cdims[i] = cshapes_ptr; 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()