Searched refs:tensorBuffer (Results 1 – 3 of 3) sorted by relevance
| /external/tflite-support/tensorflow_lite_support/java/src/java/org/tensorflow/lite/support/label/ |
| D | TensorLabel.java | 60 private final TensorBuffer tensorBuffer; field in TensorLabel
76 @NonNull Map<Integer, List<String>> axisLabels, @NonNull TensorBuffer tensorBuffer) { in TensorLabel() argument
78 SupportPreconditions.checkNotNull(tensorBuffer, "Tensor Buffer cannot be null."); in TensorLabel()
80 this.tensorBuffer = tensorBuffer; in TensorLabel()
81 this.shape = tensorBuffer.getShape(); in TensorLabel()
104 public TensorLabel(@NonNull List<String> axisLabels, @NonNull TensorBuffer tensorBuffer) { in TensorLabel() argument
105 this(makeMap(getFirstAxisWithSizeGreaterThanOne(tensorBuffer), axisLabels), tensorBuffer); in TensorLabel()
114 int labeledAxis = getFirstAxisWithSizeGreaterThanOne(tensorBuffer); in getMapWithTensorBuffer()
122 DataType dataType = tensorBuffer.getDataType(); in getMapWithTensorBuffer()
123 int typeSize = tensorBuffer.getTypeSize(); in getMapWithTensorBuffer()
[all …]
|
| D | LabelUtil.java | 45 @NonNull TensorBuffer tensorBuffer, @NonNull List<String> labels, int offset) { in mapValueToLabels()
46 SupportPreconditions.checkNotNull(tensorBuffer, "Given tensor should not be null"); in mapValueToLabels()
48 int[] values = tensorBuffer.getIntArray(); in mapValueToLabels()
|
| /external/tflite-support/tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/vision/segmenter/ |
| D | OutputType.java | 71 TensorBuffer tensorBuffer = TensorBuffer.createDynamic(UINT8); in createMasksFromBuffer() local
72 tensorBuffer.loadBuffer(buffers.get(0), maskShape); in createMasksFromBuffer()
74 tensorImage.load(tensorBuffer, GRAYSCALE); in createMasksFromBuffer()
114 TensorBuffer tensorBuffer = TensorBuffer.createDynamic(FLOAT32); in createMasksFromBuffer() local
115 tensorBuffer.loadBuffer(buffer, maskShape); in createMasksFromBuffer()
117 tensorImage.load(tensorBuffer, GRAYSCALE); in createMasksFromBuffer()
|