• Home
  • Raw
  • Download

Lines Matching refs:output

293                     Shape* output) {  in reshapePrepare()  argument
321 output->type = input.type; in reshapePrepare()
322 output->dimensions = outDims; in reshapePrepare()
323 output->offset = input.offset; in reshapePrepare()
324 output->scale = input.scale; in reshapePrepare()
329 bool depthToSpacePrepare(const Shape& input, int32_t blockSize, Shape* output) { in depthToSpacePrepare() argument
339 output->type = input.type; in depthToSpacePrepare()
340 output->dimensions = {batches, height * blockSize, width * blockSize, in depthToSpacePrepare()
342 output->offset = input.offset; in depthToSpacePrepare()
343 output->scale = input.scale; in depthToSpacePrepare()
348 bool spaceToDepthPrepare(const Shape& input, int32_t blockSize, Shape* output) { in spaceToDepthPrepare() argument
360 output->type = input.type; in spaceToDepthPrepare()
361 output->dimensions = {batches, height / blockSize, width / blockSize, in spaceToDepthPrepare()
363 output->offset = input.offset; in spaceToDepthPrepare()
364 output->scale = input.scale; in spaceToDepthPrepare()
411 Shape* output) { in padPrepare() argument
428 output->type = input.type; in padPrepare()
429 output->dimensions = outDims; in padPrepare()
430 output->offset = input.offset; in padPrepare()
431 output->scale = input.scale; in padPrepare()
437 const Shape& blockSizeShape, Shape* output) { in batchToSpacePrepare() argument
453 output->type = input.type; in batchToSpacePrepare()
454 output->dimensions = {batches / (blockSizeData[0] * blockSizeData[1]), in batchToSpacePrepare()
456 output->offset = input.offset; in batchToSpacePrepare()
457 output->scale = input.scale; in batchToSpacePrepare()
464 const Shape& paddingsShape, Shape* output) { in spaceToBatchPrepare() argument
491 output->type = input.type; in spaceToBatchPrepare()
492 output->dimensions = {batches * (blockSizeData[0] * blockSizeData[1]), in spaceToBatchPrepare()
495 output->offset = input.offset; in spaceToBatchPrepare()
496 output->scale = input.scale; in spaceToBatchPrepare()
502 Shape* output) { in meanPrepare() argument
527 output->dimensions = outDims; in meanPrepare()
568 output->dimensions = outDims; in meanPrepare()
571 output->type = input.type; in meanPrepare()
572 output->offset = input.offset; in meanPrepare()
573 output->scale = input.scale; in meanPrepare()
578 bool argMinMaxPrepare(const Shape& input, int32_t axis, Shape* output) { in argMinMaxPrepare() argument
581 output->type = OperandType::TENSOR_INT32; in argMinMaxPrepare()
584 output->dimensions.clear(); in argMinMaxPrepare()
586 output->dimensions.reserve(getNumberOfDimensions(input) - 1); in argMinMaxPrepare()
587 output->dimensions.insert(output->dimensions.end(), input.dimensions.begin(), in argMinMaxPrepare()
589 output->dimensions.insert(output->dimensions.end(), input.dimensions.begin() + axis + 1, in argMinMaxPrepare()
592 output->dimensions.push_back(1); in argMinMaxPrepare()
599 std::vector<Shape>* output) { in splitPrepare() argument
607 output->at(i).type = input.type; in splitPrepare()
608 output->at(i).dimensions = input.dimensions; in splitPrepare()
609 output->at(i).dimensions[axis] = sliceSize; in splitPrepare()
610 output->at(i).offset = input.offset; in splitPrepare()
611 output->at(i).scale = input.scale; in splitPrepare()
619 int32_t numGroups, Shape* output) { in groupedConvPrepare() argument
658 output->type = input.type; in groupedConvPrepare()
659 output->dimensions = {batches, outHeight, outWidth, channels_out}; in groupedConvPrepare()