HomeSort by relevance Sort by last modified time
    Searched refs:elemByteSize (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/
Tensor.java 144 int byteSize = elemByteSize(t.dtype) * numElements(t.shapeCopy);
266 int elemBytes = elemByteSize(dtype);
310 nbytes = nflattened * elemByteSize(dataType);
597 private static int elemByteSize(DataType dataType) {
  /external/tensorflow/tensorflow/java/src/main/native/
tensor_jni.cc 38 size_t elemByteSize(TF_DataType dtype) {
71 size_t sz = elemByteSize(dtype);
123 size_t to_copy = nelems * elemByteSize(dtype); \
155 const size_t sz = len * elemByteSize(dtype);
551 memcpy(&ret, TF_TensorData(t), elemByteSize(dtype)); \

Completed in 224 milliseconds