Searched refs:GetTensorInfoForOperand (Results 1 – 12 of 12) sorted by relevance
206 const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*outputOperand); in ConvertAdd()293 const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertArgMinMax()393 const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertBatchMatMul()456 const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertBatchToSpaceNd()551 const TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertCast()612 const TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertComparison()690 armnn::TensorInfo outputInfo = GetTensorInfoForOperand(*outputOperand); in ConvertConcatenation()1038 const TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertConv2d()1220 const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertDepthToSpace()1291 const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertDepthwiseConv2d()[all …]
81 const armnn::TensorInfo& tensor = GetTensorInfoForOperand(operand); in Convert()89 outputSlot.SetTensorInfo(GetTensorInfoForOperand(operand)); in Convert()179 const armnn::TensorInfo& tensor = GetTensorInfoForOperand(operand); in Convert()
187 armnn::TensorInfo tensorInfo = GetTensorInfoForOperand(operand); in ConvertOperandToConstTensorPin()227 armnn::TensorInfo operandTensorInfo = GetTensorInfoForOperand(*operand); in ConvertToLayerInputHandle()402 const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertPooling2d()528 const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertReduce()610 const armnn::TensorInfo& outInfo = GetTensorInfoForOperand(*outputOperand); in ConvertToActivation()983 outputSlot.SetTensorInfo(GetTensorInfoForOperand(*outputOperand)); in SetupAndTrackLayerOutputSlot()
50 armnn::TensorInfo GetTensorInfoForOperand(const Operand& operand);
97 armnn::TensorInfo GetTensorInfoForOperand(const Operand& operand) in GetTensorInfoForOperand() function
162 const TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertCast()250 const TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertChannelShuffle()317 const TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertComparison_1_2()391 const TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertConv2d_1_2()562 const TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertDepthwiseConv2d_1_2()752 const TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertElementwiseUnary()818 const TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertExpandDims()912 const TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertGather()1001 TensorInfo outputInfo = GetTensorInfoForOperand(*output); in ConvertGroupedConv2d()1374 const TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertInstanceNormalization()[all …]
886 armnn::TensorInfo tensorInfo = GetTensorInfoForOperand(operand); in ConvertOperandToConstTensorPin()1230 armnn::TensorInfo operandTensorInfo = GetTensorInfoForOperand(*operand); in ConvertToLayerInputHandle()1350 armnn::TensorInfo operandTensorInfo = GetTensorInfoForOperand(*operand); in ConvertToLayerInputHandle()1483 outputSlot.SetTensorInfo(GetTensorInfoForOperand(*outputOperand)); in SetupAndTrackLayerOutputSlot()1623 const armnn::TensorInfo& outInfo = GetTensorInfoForOperand(*outputOperand); in ConvertToActivation()1784 const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertPooling2d()1935 const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertArgMinMax()2009 armnn::TensorInfo outputInfo = GetTensorInfoForOperand(*outputOperand); in ConvertConcatenation()2376 const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertConv2d()2513 const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertDepthToSpace()[all …]
89 const TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertFill()188 const TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertLogicalBinary()631 TensorInfo outputStateOutInfo = GetTensorInfoForOperand(*outputStateOut); in ConvertQuantizedLstm()632 TensorInfo outputInfo = GetTensorInfoForOperand(*output); in ConvertQuantizedLstm()633 const TensorInfo& cellStateOutInfo = GetTensorInfoForOperand(*cellStateOut); in ConvertQuantizedLstm()774 armnn::TensorInfo outInfo = GetTensorInfoForOperand(*outputOperand); in ConvertRank()
73 armnn::TensorInfo GetTensorInfoForOperand(const V1_0::Operand& operand);81 armnn::TensorInfo GetTensorInfoForOperand(const V1_2::Operand& operand);89 armnn::TensorInfo GetTensorInfoForOperand(const V1_3::Operand& operand);
93 armnn::TensorInfo GetTensorInfoForOperand(const V1_0::Operand& operand) in GetTensorInfoForOperand() function134 armnn::TensorInfo GetTensorInfoForOperand(const V1_2::Operand& operand) in GetTensorInfoForOperand() function208 armnn::TensorInfo GetTensorInfoForOperand(const V1_3::Operand& operand) in GetTensorInfoForOperand() function
104 outputSlot.SetTensorInfo(GetTensorInfoForOperand(operand)); in Convert()
425 const armnn::TensorInfo& scratchBufferInfo = GetTensorInfoForOperand(*scratchBuffer); in ConvertLstm()426 const armnn::TensorInfo& outputStateOutInfo = GetTensorInfoForOperand(*outputStateOut); in ConvertLstm()427 const armnn::TensorInfo& cellStateOutInfo = GetTensorInfoForOperand(*cellStateOut); in ConvertLstm()428 const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertLstm()556 const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*outputOperand); in ConvertSoftmax()627 const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertSpaceToDepth()688 const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*output); in ConvertResizeBilinear()