Lines Matching refs:shape
85 bool setOutputShape(uint32_t index, const Shape& shape) override;
127 return getInputInfo(index)->shape(); in getInputShape()
143 return getOutputInfo(index)->shape(); in getOutputShape()
168 bool setInfoAndAllocateIfNeeded(RunTimeOperandInfo* info, const Shape& shape, int* result) { in setInfoAndAllocateIfNeeded() argument
172 if (info->type != shape.type) { in setInfoAndAllocateIfNeeded()
177 if (info->scale != shape.scale) { in setInfoAndAllocateIfNeeded()
182 if (info->zeroPoint != shape.offset) { in setInfoAndAllocateIfNeeded()
187 if (info->extraParams != shape.extraParams) { in setInfoAndAllocateIfNeeded()
194 auto combined = combineDimensions(shape.dimensions, info->dimensions); in setInfoAndAllocateIfNeeded()
201 info->type = shape.type; in setInfoAndAllocateIfNeeded()
202 info->scale = shape.scale; in setInfoAndAllocateIfNeeded()
203 info->zeroPoint = shape.offset; in setInfoAndAllocateIfNeeded()
204 info->extraParams = shape.extraParams; in setInfoAndAllocateIfNeeded()
244 bool OperationExecutionContext::setOutputShape(uint32_t index, const Shape& shape) { in setOutputShape() argument
245 return setInfoAndAllocateIfNeeded(getOutputInfo(index), shape, &result); in setOutputShape()
444 Shape inShape = from.shape(); in convertToNhwc()
486 Shape outShape = from.shape(); in convertFromNhwc()
514 Shape outShape = from.shape(); in convertFromNhwc()
802 Shape outShape = output.shape(); in executeOperation()
804 success = reshapePrepare(input.shape(), in executeOperation()
806 getNumberOfElements(targetShape.shape()), &outShape) && in executeOperation()
808 copyData(input.buffer, input.shape(), output.buffer, outShape); in executeOperation()
820 Shape outShape = output.shape(); in executeOperation()
831 if (!depthToSpacePrepare(input_tmp.shape(), blockSize, &outShape) || in executeOperation()
839 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
845 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
851 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
857 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
884 Shape outShape = output.shape(); in executeOperation()
896 if (!spaceToDepthPrepare(input_tmp.shape(), blockSize, &outShape) || in executeOperation()
904 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
910 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
916 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
922 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
950 success = embeddingLookupPrepare(values.shape(), lookups.shape(), &outputShape) && in executeOperation()
967 success = hashtableLookupPrepare(lookups.shape(), keys.shape(), values.shape(), in executeOperation()
1105 Shape outShape = output.shape(); in executeOperation()
1117 if (!batchToSpacePrepare(input_tmp.shape(), in executeOperation()
1119 blockSize.shape(), &outShape) || in executeOperation()
1127 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1134 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1141 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1148 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1177 Shape outShape = output.shape(); in executeOperation()
1190 input_tmp.shape(), reinterpret_cast<const int32_t*>(blockSize.buffer), in executeOperation()
1191 blockSize.shape(), reinterpret_cast<const int32_t*>(paddings.buffer), in executeOperation()
1192 paddings.shape(), &outShape) || in executeOperation()
1200 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1202 reinterpret_cast<const int32_t*>(paddings.buffer), paddings.shape(), in executeOperation()
1208 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1210 reinterpret_cast<const int32_t*>(paddings.buffer), paddings.shape(), in executeOperation()
1216 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1218 reinterpret_cast<const int32_t*>(paddings.buffer), paddings.shape(), in executeOperation()
1224 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1226 reinterpret_cast<const int32_t*>(paddings.buffer), paddings.shape(), in executeOperation()
1253 Shape outShape = output.shape(); in executeOperation()
1255 if (!padPrepare(input.shape(), reinterpret_cast<const int32_t*>(paddings.buffer), in executeOperation()
1256 paddings.shape(), &outShape) || in executeOperation()
1262 success = padGeneric(reinterpret_cast<const float*>(input.buffer), input.shape(), in executeOperation()
1267 success = padGeneric(reinterpret_cast<const _Float16*>(input.buffer), input.shape(), in executeOperation()
1274 success = padGeneric(input.buffer, input.shape(), in executeOperation()
1280 success = padGeneric(input.buffer, input.shape(), in executeOperation()
1292 Shape outShape = output.shape(); in executeOperation()
1294 success = cast::prepare(input.shape(), &outShape) && in executeOperation()
1296 cast::eval(input.buffer, input.shape(), output.buffer, outShape); in executeOperation()
1307 Shape outShape = output.shape(); in executeOperation()
1309 if (!meanPrepare(input.shape(), reinterpret_cast<const int32_t*>(axis.buffer), in executeOperation()
1310 axis.shape(), keepDims > 0, &outShape) || in executeOperation()
1315 success = meanFloat16(reinterpret_cast<_Float16*>(input.buffer), input.shape(), in executeOperation()
1316 reinterpret_cast<const int32_t*>(axis.buffer), axis.shape(), in executeOperation()
1321 reinterpret_cast<float*>(input.buffer), input.shape(), in executeOperation()
1322 reinterpret_cast<const int32_t*>(axis.buffer), axis.shape(), keepDims > 0, in executeOperation()
1326 reinterpret_cast<uint8_t*>(input.buffer), input.shape(), in executeOperation()
1327 reinterpret_cast<const int32_t*>(axis.buffer), axis.shape(), keepDims > 0, in executeOperation()
1331 reinterpret_cast<int8_t*>(input.buffer), input.shape(), in executeOperation()
1332 reinterpret_cast<const int32_t*>(axis.buffer), axis.shape(), keepDims > 0, in executeOperation()
1345 Shape outShape = output.shape(); in executeOperation()
1348 success = argMinMaxPrepare(input.shape(), axis, &outShape) && in executeOperation()
1350 argMinMaxGeneric(input.buffer, input.shape(), axis, isArgMin, output.buffer, in executeOperation()
1361 Shape outShape = output.shape(); in executeOperation()
1363 success = expand_dims::prepare(input.shape(), axis, &outShape) && in executeOperation()
1365 expand_dims::eval(input.buffer, input.shape(), axis, output.buffer, outShape); in executeOperation()
1383 outputShapes[i] = operands[outs[i]].shape(); in executeOperation()
1386 success = splitPrepare(input.shape(), axis, numOutputs, &outputShapes); in executeOperation()
1399 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1408 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1417 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1426 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1435 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1451 Shape outputShape = output.shape(); in executeOperation()
1454 success = maximum_minimum::prepare(in1.shape(), in2.shape(), &outputShape) && in executeOperation()
1456 maximum_minimum::eval(in1.buffer, in1.shape(), in2.buffer, in2.shape(), in executeOperation()
1496 Shape outShape = output.shape(); in executeOperation()
1509 Shape inputShape = input_tmp.shape(); in executeOperation()
1510 Shape filterShape = filter.shape(); in executeOperation()
1521 if (!groupedConvPrepare(input_tmp.shape(), filter.shape(), bias.shape(), padding_left, in executeOperation()
1532 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1533 reinterpret_cast<const float*>(filter.buffer), filter.shape(), in executeOperation()
1534 reinterpret_cast<const float*>(bias.buffer), bias.shape(), padding_left, in executeOperation()
1540 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1541 reinterpret_cast<const _Float16*>(filter.buffer), filter.shape(), in executeOperation()
1542 reinterpret_cast<const _Float16*>(bias.buffer), bias.shape(), padding_left, in executeOperation()
1549 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1550 reinterpret_cast<const int8_t*>(filter.buffer), filter.shape(), in executeOperation()
1553 reinterpret_cast<const int32_t*>(bias.buffer), bias.shape(), in executeOperation()
1559 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1560 reinterpret_cast<const uint8_t*>(filter.buffer), filter.shape(), in executeOperation()
1561 reinterpret_cast<const int32_t*>(bias.buffer), bias.shape(), in executeOperation()
1569 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1570 reinterpret_cast<const int8_t*>(filter.buffer), filter.shape(), in executeOperation()
1573 reinterpret_cast<const int32_t*>(bias.buffer), bias.shape(), in executeOperation()
1579 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1580 reinterpret_cast<const int8_t*>(filter.buffer), filter.shape(), in executeOperation()
1581 reinterpret_cast<const int32_t*>(bias.buffer), bias.shape(), in executeOperation()
1604 Shape outShape = output.shape(); in executeOperation()
1607 tile::prepare(input.shape(), reinterpret_cast<const int32_t*>(multiples.buffer), in executeOperation()
1608 multiples.shape(), &outShape) && in executeOperation()
1610 tile::eval(input.buffer, input.shape(), in executeOperation()
1640 Shape outShape = output.shape(); in executeOperation()
1642 success = pow::prepare(base.shape(), exponent.shape(), &outShape) && in executeOperation()
1644 pow::eval(base.buffer, base.shape(), exponent.buffer, exponent.shape(), in executeOperation()
1849 if (int error; !setInfoAndAllocateIfNeeded(&outerOperand, innerOperand.shape(), &error)) { in executeWhileOperation()