Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/java/src/main/native/
Dtensor_jni.cc38 size_t elemByteSize(TF_DataType dtype) { in elemByteSize() function
71 size_t sz = elemByteSize(dtype); in writeScalar()
123 size_t to_copy = nelems * elemByteSize(dtype); \ in write1DArray()
155 const size_t sz = len * elemByteSize(dtype); in read1DArray()
500 memcpy(&ret, TF_TensorData(t), elemByteSize(dtype)); \
/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/
DTensor.java145 int byteSize = elemByteSize(t.dtype) * numElements(t.shapeCopy); in create()
253 int elemBytes = elemByteSize(dtype); in create()
297 nbytes = nflattened * elemByteSize(dataType); in allocateForBuffer()
594 private static int elemByteSize(DataType dataType) { in elemByteSize() method in Tensor