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

  /external/tensorflow/tensorflow/contrib/lite/java/src/test/java/org/tensorflow/lite/
DataTypeTest.java 29 assertThat(DataType.FLOAT32.elemByteSize()).isEqualTo(4);
30 assertThat(DataType.INT32.elemByteSize()).isEqualTo(4);
31 assertThat(DataType.UINT8.elemByteSize()).isEqualTo(1);
32 assertThat(DataType.INT64.elemByteSize()).isEqualTo(8);
  /external/tensorflow/tensorflow/contrib/lite/java/src/main/java/org/tensorflow/lite/
DataType.java 58 int elemByteSize() {
NativeInterpreterWrapper.java 86 numsOfBytes[i] = dataType.elemByteSize() * numElements(dims);
  /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);
544 memcpy(&ret, TF_TensorData(t), elemByteSize(dtype)); \

Completed in 239 milliseconds