Home
last modified time | relevance | path

Searched refs:num_dims_elems (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/java/src/main/native/
Deager_operation_builder_jni.cc317 jint* num_dims_elems = in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList() local
320 cnum_dims[i] = static_cast<int>(num_dims_elems[i]); in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList()
326 env->ReleasePrimitiveArrayCritical(num_dims, num_dims_elems, JNI_ABORT); in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList()
Dgraph_operation_builder_jni.cc291 jint* num_dims_elems = in Java_org_tensorflow_GraphOperationBuilder_setAttrShapeList() local
294 cnum_dims[i] = static_cast<int>(num_dims_elems[i]); in Java_org_tensorflow_GraphOperationBuilder_setAttrShapeList()
300 env->ReleasePrimitiveArrayCritical(num_dims, num_dims_elems, JNI_ABORT); in Java_org_tensorflow_GraphOperationBuilder_setAttrShapeList()