Lines Matching refs:shape
71 bool setOutputShape(uint32_t index, const Shape& shape) override;
113 return getInputInfo(index)->shape(); in getInputShape()
129 return getOutputInfo(index)->shape(); in getOutputShape()
154 bool setInfoAndAllocateIfNeeded(RunTimeOperandInfo* info, const Shape& shape, int* result) { in setInfoAndAllocateIfNeeded() argument
158 if (info->type != shape.type) { in setInfoAndAllocateIfNeeded()
163 if (info->scale != shape.scale) { in setInfoAndAllocateIfNeeded()
168 if (info->zeroPoint != shape.offset) { in setInfoAndAllocateIfNeeded()
173 if (info->extraParams != shape.extraParams) { in setInfoAndAllocateIfNeeded()
180 auto combined = combineDimensions(shape.dimensions, info->dimensions); in setInfoAndAllocateIfNeeded()
187 info->type = shape.type; in setInfoAndAllocateIfNeeded()
188 info->scale = shape.scale; in setInfoAndAllocateIfNeeded()
189 info->zeroPoint = shape.offset; in setInfoAndAllocateIfNeeded()
190 info->extraParams = shape.extraParams; in setInfoAndAllocateIfNeeded()
230 bool OperationExecutionContext::setOutputShape(uint32_t index, const Shape& shape) { in setOutputShape() argument
231 return setInfoAndAllocateIfNeeded(getOutputInfo(index), shape, &result); in setOutputShape()
527 Shape inShape = from.shape(); in convertToNhwc()
569 Shape outShape = from.shape(); in convertFromNhwc()
597 Shape outShape = from.shape(); in convertFromNhwc()
858 Shape outShape = output.shape(); in executeOperation()
860 success = reshapePrepare(input.shape(), in executeOperation()
862 getNumberOfElements(targetShape.shape()), &outShape) && in executeOperation()
864 copyData(input.buffer, input.shape(), output.buffer, outShape); in executeOperation()
876 Shape outShape = output.shape(); in executeOperation()
887 if (!depthToSpacePrepare(input_tmp.shape(), blockSize, &outShape) || in executeOperation()
895 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
901 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
907 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
913 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
940 Shape outShape = output.shape(); in executeOperation()
952 if (!spaceToDepthPrepare(input_tmp.shape(), blockSize, &outShape) || in executeOperation()
960 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
966 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
972 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
978 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1006 success = embeddingLookupPrepare(values.shape(), lookups.shape(), &outputShape) && in executeOperation()
1023 success = hashtableLookupPrepare(lookups.shape(), keys.shape(), values.shape(), in executeOperation()
1161 Shape outShape = output.shape(); in executeOperation()
1173 if (!batchToSpacePrepare(input_tmp.shape(), in executeOperation()
1175 blockSize.shape(), &outShape) || in executeOperation()
1183 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1190 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1197 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1204 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1233 Shape outShape = output.shape(); in executeOperation()
1246 input_tmp.shape(), reinterpret_cast<const int32_t*>(blockSize.buffer), in executeOperation()
1247 blockSize.shape(), reinterpret_cast<const int32_t*>(paddings.buffer), in executeOperation()
1248 paddings.shape(), &outShape) || in executeOperation()
1256 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1258 reinterpret_cast<const int32_t*>(paddings.buffer), paddings.shape(), in executeOperation()
1264 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1266 reinterpret_cast<const int32_t*>(paddings.buffer), paddings.shape(), in executeOperation()
1272 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1274 reinterpret_cast<const int32_t*>(paddings.buffer), paddings.shape(), in executeOperation()
1280 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1282 reinterpret_cast<const int32_t*>(paddings.buffer), paddings.shape(), in executeOperation()
1309 Shape outShape = output.shape(); in executeOperation()
1311 if (!padPrepare(input.shape(), reinterpret_cast<const int32_t*>(paddings.buffer), in executeOperation()
1312 paddings.shape(), &outShape) || in executeOperation()
1318 success = padGeneric(reinterpret_cast<const float*>(input.buffer), input.shape(), in executeOperation()
1323 success = padGeneric(reinterpret_cast<const _Float16*>(input.buffer), input.shape(), in executeOperation()
1330 success = padGeneric(input.buffer, input.shape(), in executeOperation()
1336 success = padGeneric(input.buffer, input.shape(), in executeOperation()
1348 Shape outShape = output.shape(); in executeOperation()
1350 success = cast::prepare(input.shape(), &outShape) && in executeOperation()
1352 cast::eval(input.buffer, input.shape(), output.buffer, outShape); in executeOperation()
1363 Shape outShape = output.shape(); in executeOperation()
1365 if (!meanPrepare(input.shape(), reinterpret_cast<const int32_t*>(axis.buffer), in executeOperation()
1366 axis.shape(), keepDims > 0, &outShape) || in executeOperation()
1371 success = meanFloat16(reinterpret_cast<_Float16*>(input.buffer), input.shape(), in executeOperation()
1372 reinterpret_cast<const int32_t*>(axis.buffer), axis.shape(), in executeOperation()
1377 reinterpret_cast<float*>(input.buffer), input.shape(), in executeOperation()
1378 reinterpret_cast<const int32_t*>(axis.buffer), axis.shape(), keepDims > 0, in executeOperation()
1382 reinterpret_cast<uint8_t*>(input.buffer), input.shape(), in executeOperation()
1383 reinterpret_cast<const int32_t*>(axis.buffer), axis.shape(), keepDims > 0, in executeOperation()
1387 reinterpret_cast<int8_t*>(input.buffer), input.shape(), in executeOperation()
1388 reinterpret_cast<const int32_t*>(axis.buffer), axis.shape(), keepDims > 0, in executeOperation()
1401 Shape outShape = output.shape(); in executeOperation()
1404 success = argMinMaxPrepare(input.shape(), axis, &outShape) && in executeOperation()
1406 argMinMaxGeneric(input.buffer, input.shape(), axis, isArgMin, output.buffer, in executeOperation()
1417 Shape outShape = output.shape(); in executeOperation()
1419 success = expand_dims::prepare(input.shape(), axis, &outShape) && in executeOperation()
1421 expand_dims::eval(input.buffer, input.shape(), axis, output.buffer, outShape); in executeOperation()
1439 outputShapes[i] = operands[outs[i]].shape(); in executeOperation()
1442 success = splitPrepare(input.shape(), axis, numOutputs, &outputShapes); in executeOperation()
1455 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1464 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1473 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1482 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1491 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1507 Shape outputShape = output.shape(); in executeOperation()
1510 success = maximum_minimum::prepare(in1.shape(), in2.shape(), &outputShape) && in executeOperation()
1512 maximum_minimum::eval(in1.buffer, in1.shape(), in2.buffer, in2.shape(), in executeOperation()
1552 Shape outShape = output.shape(); in executeOperation()
1565 Shape inputShape = input_tmp.shape(); in executeOperation()
1566 Shape filterShape = filter.shape(); in executeOperation()
1577 if (!groupedConvPrepare(input_tmp.shape(), filter.shape(), bias.shape(), padding_left, in executeOperation()
1588 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1589 reinterpret_cast<const float*>(filter.buffer), filter.shape(), in executeOperation()
1590 reinterpret_cast<const float*>(bias.buffer), bias.shape(), padding_left, in executeOperation()
1596 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1597 reinterpret_cast<const _Float16*>(filter.buffer), filter.shape(), in executeOperation()
1598 reinterpret_cast<const _Float16*>(bias.buffer), bias.shape(), padding_left, in executeOperation()
1605 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1606 reinterpret_cast<const int8_t*>(filter.buffer), filter.shape(), in executeOperation()
1608 reinterpret_cast<const int32_t*>(bias.buffer), bias.shape(), in executeOperation()
1614 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1615 reinterpret_cast<const uint8_t*>(filter.buffer), filter.shape(), in executeOperation()
1616 reinterpret_cast<const int32_t*>(bias.buffer), bias.shape(), in executeOperation()
1624 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1625 reinterpret_cast<const int8_t*>(filter.buffer), filter.shape(), in executeOperation()
1627 reinterpret_cast<const int32_t*>(bias.buffer), bias.shape(), in executeOperation()
1633 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1634 reinterpret_cast<const int8_t*>(filter.buffer), filter.shape(), in executeOperation()
1635 reinterpret_cast<const int32_t*>(bias.buffer), bias.shape(), in executeOperation()
1658 Shape outShape = output.shape(); in executeOperation()
1661 tile::prepare(input.shape(), reinterpret_cast<const int32_t*>(multiples.buffer), in executeOperation()
1662 multiples.shape(), &outShape) && in executeOperation()
1664 tile::eval(input.buffer, input.shape(), in executeOperation()
1694 Shape outShape = output.shape(); in executeOperation()
1696 success = pow::prepare(base.shape(), exponent.shape(), &outShape) && in executeOperation()
1698 pow::eval(base.buffer, base.shape(), exponent.buffer, exponent.shape(), in executeOperation()
1898 if (int error; !setInfoAndAllocateIfNeeded(&outerOperand, innerOperand.shape(), &error)) { in executeWhileOperation()