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() 483 writeNDArray(env, static_cast<jarray>(value), dtype, num_dims, in Java_org_tensorflow_Tensor_setValue()
|