Searched refs:flatSize (Results 1 – 5 of 5) sorted by relevance
37 protected int flatSize = -1; field in TensorBuffer149 return flatSize; in getFlatSize()308 int flatSize = computeFlatSize(shape); in loadBuffer() local310 (buffer.limit() == getTypeSize() * flatSize), in loadBuffer()387 if (flatSize == newFlatSize) { in allocateMemory()392 flatSize = newFlatSize; in allocateMemory()393 buffer = ByteBuffer.allocateDirect(flatSize * getTypeSize()); in allocateMemory()402 int flatSize = computeFlatSize(shape); in assertShapeIsCorect() local404 (buffer.limit() == getTypeSize() * flatSize), in assertShapeIsCorect()
49 byte[] byteArr = new byte[flatSize]; in getFloatArray()52 float[] floatArr = new float[flatSize]; in getFloatArray()53 for (int i = 0; i < flatSize; i++) { in getFloatArray()68 byte[] byteArr = new byte[flatSize]; in getIntArray()71 int[] intArr = new int[flatSize]; in getIntArray()72 for (int i = 0; i < flatSize; i++) { in getIntArray()
50 float[] arr = new float[flatSize]; in getFloatArray()66 float[] floatArr = new float[flatSize]; in getIntArray()69 int[] intArr = new int[flatSize]; in getIntArray()70 for (int i = 0; i < flatSize; i++) { in getIntArray()
113 int flatSize = w * h * 3; in convertBitmapToTensorBuffer() local123 ByteBuffer byteBuffer = ByteBuffer.allocateDirect(flatSize); in convertBitmapToTensorBuffer()
124 int flatSize = tensorBuffer.getFlatSize(); in getMapWithTensorBuffer() local132 int subArrayLength = flatSize / shape[labeledAxis] * typeSize; in getMapWithTensorBuffer()