Lines Matching refs:outShape
569 Shape outShape = from.shape(); in convertFromNhwc() local
571 outShape.dimensions = {fromDim[0], fromDim[3], fromDim[1], fromDim[2]}; in convertFromNhwc()
573 if (!setInfoAndAllocateIfNeeded(&to, outShape, result)) { in convertFromNhwc()
597 Shape outShape = from.shape(); in convertFromNhwc() local
600 if (!setInfoAndAllocateIfNeeded(&to, outShape, result)) { in convertFromNhwc()
858 Shape outShape = output.shape(); in executeOperation() local
862 getNumberOfElements(targetShape.shape()), &outShape) && in executeOperation()
863 setInfoAndAllocateIfNeeded(&output, outShape, &result) && in executeOperation()
864 copyData(input.buffer, input.shape(), output.buffer, outShape); in executeOperation()
876 Shape outShape = output.shape(); in executeOperation() local
887 if (!depthToSpacePrepare(input_tmp.shape(), blockSize, &outShape) || in executeOperation()
888 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
896 blockSize, reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
902 blockSize, reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
908 blockSize, reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
914 blockSize, reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
940 Shape outShape = output.shape(); in executeOperation() local
952 if (!spaceToDepthPrepare(input_tmp.shape(), blockSize, &outShape) || in executeOperation()
953 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
961 blockSize, reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
967 blockSize, reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
973 blockSize, reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
979 blockSize, reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
1161 Shape outShape = output.shape(); in executeOperation() local
1175 blockSize.shape(), &outShape) || in executeOperation()
1176 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
1185 reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
1192 reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
1199 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1206 reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
1233 Shape outShape = output.shape(); in executeOperation() local
1248 paddings.shape(), &outShape) || in executeOperation()
1249 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
1259 reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
1267 reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
1275 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1283 reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
1309 Shape outShape = output.shape(); in executeOperation() local
1312 paddings.shape(), &outShape) || in executeOperation()
1313 !setInfoAndAllocateIfNeeded(&output, outShape, &result)) { in executeOperation()
1320 reinterpret_cast<float*>(output.buffer), outShape); in executeOperation()
1326 reinterpret_cast<_Float16*>(output.buffer), outShape); in executeOperation()
1329 isV2 ? getScalarData<uint8_t>(operands[ins[2]]) : outShape.offset; in executeOperation()
1332 output.buffer, outShape); in executeOperation()
1335 isV2 ? getScalarData<int8_t>(operands[ins[2]]) : outShape.offset; in executeOperation()
1338 output.buffer, outShape); in executeOperation()
1348 Shape outShape = output.shape(); in executeOperation() local
1350 success = cast::prepare(input.shape(), &outShape) && in executeOperation()
1351 setInfoAndAllocateIfNeeded(&output, outShape, &result) && in executeOperation()
1352 cast::eval(input.buffer, input.shape(), output.buffer, outShape); in executeOperation()
1363 Shape outShape = output.shape(); in executeOperation() local
1366 axis.shape(), keepDims > 0, &outShape) || in executeOperation()
1367 !setInfoAndAllocateIfNeeded(&output, outShape, &result)) { in executeOperation()
1374 outShape); in executeOperation()
1379 reinterpret_cast<float*>(output.buffer), outShape); in executeOperation()
1384 reinterpret_cast<uint8_t*>(output.buffer), outShape); in executeOperation()
1389 reinterpret_cast<int8_t*>(output.buffer), outShape); in executeOperation()
1401 Shape outShape = output.shape(); in executeOperation() local
1404 success = argMinMaxPrepare(input.shape(), axis, &outShape) && in executeOperation()
1405 setInfoAndAllocateIfNeeded(&output, outShape, &result) && in executeOperation()
1407 outShape); in executeOperation()
1417 Shape outShape = output.shape(); in executeOperation() local
1419 success = expand_dims::prepare(input.shape(), axis, &outShape) && in executeOperation()
1420 setInfoAndAllocateIfNeeded(&output, outShape, &result) && in executeOperation()
1421 expand_dims::eval(input.buffer, input.shape(), axis, output.buffer, outShape); in executeOperation()
1552 Shape outShape = output.shape(); in executeOperation() local
1579 stride_height, numGroups, &outShape) || in executeOperation()
1580 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
1593 outShape); in executeOperation()
1601 outShape); in executeOperation()
1611 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1619 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1630 reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
1638 reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
1658 Shape outShape = output.shape(); in executeOperation() local
1662 multiples.shape(), &outShape) && in executeOperation()
1663 setInfoAndAllocateIfNeeded(&output, outShape, &result) && in executeOperation()
1666 outShape); in executeOperation()
1694 Shape outShape = output.shape(); in executeOperation() local
1696 success = pow::prepare(base.shape(), exponent.shape(), &outShape) && in executeOperation()
1697 setInfoAndAllocateIfNeeded(&output, outShape, &result) && in executeOperation()
1699 output.buffer, outShape); in executeOperation()