• Home
  • Raw
  • Download

Lines Matching refs:SetTensorInfo

1048         inputLayer->GetOutputSlot(0).SetTensorInfo(tensorInfo);  in SetupInputLayers()
1210 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseAbs()
1238 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseActivation()
1259 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseAdd()
1285 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseArgMinMax()
1316 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseBatchMatMul()
1358 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseBatchToSpaceNd()
1395 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParseBatchNormalization()
1416 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseCast()
1466 layer->GetOutputSlot(0).SetTensorInfo(weightsPermuted.GetInfo()); in ParseConstant()
1478 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseConstant()
1528 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsTensor.GetInfo()); in ParseConvolution2d()
1536 biasLayer->GetOutputSlot(0).SetTensorInfo(biasTensor.GetInfo()); in ParseConvolution2d()
1549 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseConvolution2d()
1590 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseConvolution3d()
1616 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParseDepthToSpace()
1672 biasLayer->GetOutputSlot(0).SetTensorInfo(biases.GetInfo()); in ParseDepthwiseConvolution2d()
1698 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsPermuted.GetInfo()); in ParseDepthwiseConvolution2d()
1704 weightsLayer->GetOutputSlot(0).SetTensorInfo(weights.GetInfo()); in ParseDepthwiseConvolution2d()
1716 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseDepthwiseConvolution2d()
1757 layer->GetOutputSlot(i).SetTensorInfo(ToTensorInfo(outputs[i])); in ParseDetectionPostProcess()
1779 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseDivision()
1800 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseEqual()
1822 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseFill()
1843 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseGreater()
1872 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParseInstanceNormalization()
1898 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParseL2Normalization()
1925 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseLogicalBinary()
1949 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseLogSoftmax()
1970 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseMinimum()
1991 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseMaximum()
2031 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseChannelShuffle()
2057 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseComparison()
2084 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseElementwiseBinary()
2111 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseElementwiseUnary()
2149 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseConcat()
2170 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseMultiplication()
2194 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseFloor()
2234 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsTensor.GetInfo()); in ParseFullyConnected()
2242 biasLayer->GetOutputSlot(0).SetTensorInfo(biasTensor.GetInfo()); in ParseFullyConnected()
2255 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseFullyConnected()
2295 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParsePad()
2319 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParsePermute()
2534 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParsePooling2d()
2555 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParsePooling3d()
2574 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParseQuantize()
2624 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseRank()
2654 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseReduce()
2697 layer->GetOutputSlot(0).SetTensorInfo(reshapeOutputTensorInfo); in ParseReshape()
2727 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseResize()
2760 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseResizeBilinear()
2780 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseShape()
2804 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseSoftmax()
2847 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseSpaceToBatchNd()
2873 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseSpaceToDepth()
2966 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParseNormalization()
2987 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseRsqrt()
3022 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseSlice()
3067 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseStridedSlice()
3088 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseSubtraction()
3111 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseGather()
3131 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseGatherNd()
3159 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseMean()
3202 layer->GetOutputSlot(vIdx).SetTensorInfo(outputTensorInfo); in ParseSplitter()
3325 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo1); in ParseLstm()
3328 layer->GetOutputSlot(1).SetTensorInfo(outputTensorInfo2); in ParseLstm()
3331 layer->GetOutputSlot(2).SetTensorInfo(outputTensorInfo3); in ParseLstm()
3334 layer->GetOutputSlot(3).SetTensorInfo(outputTensorInfo4); in ParseLstm()
3477 layer->GetOutputSlot(0).SetTensorInfo(outputStateOutInfo); in ParseQLstm()
3480 layer->GetOutputSlot(1).SetTensorInfo(cellStateOutInfo); in ParseQLstm()
3483 layer->GetOutputSlot(2).SetTensorInfo(outputInfo); in ParseQLstm()
3534 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo1); in ParseQuantizedLstm()
3537 layer->GetOutputSlot(1).SetTensorInfo(outputTensorInfo2); in ParseQuantizedLstm()
3557 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseDequantize()
3577 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseMerge()
3597 layer->GetOutputSlot(0).SetTensorInfo(output0TensorInfo); in ParseSwitch()
3600 layer->GetOutputSlot(1).SetTensorInfo(output1TensorInfo); in ParseSwitch()
3620 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParsePrelu()
3643 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParseTranspose()
3688 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseTransposeConvolution2d()
3736 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseStack()
3765 layer->GetOutputSlot(i).SetTensorInfo(outputInfo); in ParseStandIn()
3896 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo0); in ParseUnidirectionalSequenceLstm()
3899 layer->GetOutputSlot(1).SetTensorInfo(outputTensorInfo1); in ParseUnidirectionalSequenceLstm()
3902 layer->GetOutputSlot(2).SetTensorInfo(outputTensorInfo2); in ParseUnidirectionalSequenceLstm()