Searched refs:labeledAxis (Results 1 – 1 of 1) sorted by relevance
114 int labeledAxis = getFirstAxisWithSizeGreaterThanOne(tensorBuffer); in getMapWithTensorBuffer() local118 axisLabels.containsKey(labeledAxis), in getMapWithTensorBuffer()120 List<String> labels = axisLabels.get(labeledAxis); in getMapWithTensorBuffer()132 int subArrayLength = flatSize / shape[labeledAxis] * typeSize; in getMapWithTensorBuffer()142 labelBuffer.loadBuffer(subBuffer, Arrays.copyOfRange(shape, labeledAxis + 1, shape.length)); in getMapWithTensorBuffer()160 int labeledAxis = getFirstAxisWithSizeGreaterThanOne(tensorBuffer); in getMapWithFloatValue() local162 labeledAxis == shape.length - 1, in getMapWithFloatValue()164 List<String> labels = axisLabels.get(labeledAxis); in getMapWithFloatValue()191 int labeledAxis = getFirstAxisWithSizeGreaterThanOne(tensorBuffer); in getCategoryList() local193 labeledAxis == shape.length - 1, in getCategoryList()[all …]