Searched refs:writeNDArray (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | tensor_jni.cc | 185 size_t writeNDArray(JNIEnv* env, jarray src, TF_DataType dtype, int dims_left, in writeNDArray() function
196 writeNDArray(env, row, dtype, dims_left - 1, dst + sz, dst_size - sz); in writeNDArray()
534 writeNDArray(env, static_cast<jarray>(value), dtype, num_dims, in Java_org_tensorflow_Tensor_setValue()
|