Searched refs:dims_left (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/java/src/main/native/ |
| D | tensor_jni.cc | 186 size_t src_size, int dims_left, jarray dst) { in ReadMultiDimensionalArray() argument
187 if (dims_left == 1) { in ReadMultiDimensionalArray()
196 src_size - size, dims_left - 1, row); in ReadMultiDimensionalArray()
206 int dims_left, int start_str_index, in ReadMultiDimensionalStringArray() argument
214 if (dims_left == 1) { in ReadMultiDimensionalStringArray()
228 env, tensor, dims_left - 1, start_str_index + num_strings_read, row); in ReadMultiDimensionalStringArray()
238 int dims_left, char** dst, int dst_size) { in WriteMultiDimensionalArray() argument
239 if (dims_left <= 1) { in WriteMultiDimensionalArray()
248 sz += WriteMultiDimensionalArray(env, row, type, dims_left - 1, &next_dst, in WriteMultiDimensionalArray()
259 int dims_left) { in PopulateStringDynamicBuffer() argument
[all …]
|
| /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() argument
187 if (dims_left == 1) { in writeNDArray()
196 writeNDArray(env, row, dtype, dims_left - 1, dst + sz, dst_size - sz); in writeNDArray()
205 size_t src_size, int dims_left, jarray dst) { in readNDArray() argument
206 if (dims_left == 1) { in readNDArray()
215 readNDArray(env, dtype, src + sz, src_size - sz, dims_left - 1, row); in readNDArray()
306 void readNDStringArray(JNIEnv* env, StringTensorReader* reader, int dims_left, in readNDStringArray() argument
309 if (dims_left == 1) { in readNDStringArray()
320 readNDStringArray(env, reader, dims_left - 1, arr, status); in readNDStringArray()
|