Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/java/src/main/native/
Doperation_builder_jni.cc273 std::unique_ptr<int[]> cnum_dims; in Java_org_tensorflow_OperationBuilder_setAttrShapeList() local
279 cnum_dims.reset(new int[num_dims_length]); in Java_org_tensorflow_OperationBuilder_setAttrShapeList()
288 cnum_dims[i] = static_cast<int>(num_dims_elems[i]); in Java_org_tensorflow_OperationBuilder_setAttrShapeList()
290 if (cnum_dims[i] > 0) { in Java_org_tensorflow_OperationBuilder_setAttrShapeList()
291 cshapes_ptr += cnum_dims[i]; 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()