Searched refs:tfLiteBlockShapeTensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/armnn/delegate/classic/src/ |
| H A D | BatchSpace.hpp | 32 const TfLiteTensor& tfLiteBlockShapeTensor = tfLiteTensors[tfLiteNode->inputs->data[1]]; in VisitBatchToSpaceNdOperator() local
33 if (!IsValid(tfLiteContext, tfLiteBlockShapeTensor, operatorCode, nodeIndex)) in VisitBatchToSpaceNdOperator()
51 …nst armnn::TensorInfo& blockShapeTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteBlockShapeTensor); in VisitBatchToSpaceNdOperator()
56 … ::memcpy(blockShape.data(), tfLiteBlockShapeTensor.data.data, blockShapeTensorInfo.GetNumBytes()); in VisitBatchToSpaceNdOperator()
132 const TfLiteTensor& tfLiteBlockShapeTensor = tfLiteTensors[tfLiteNode->inputs->data[1]]; in VisitSpaceToBatchNdOperator() local
133 if (!IsValid(tfLiteContext, tfLiteBlockShapeTensor, operatorCode, nodeIndex)) in VisitSpaceToBatchNdOperator()
151 …nst armnn::TensorInfo& blockShapeTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteBlockShapeTensor); in VisitSpaceToBatchNdOperator()
156 …std::memcpy(blockShape.data(), tfLiteBlockShapeTensor.data.data, blockShapeTensorInfo.GetNumBytes(… in VisitSpaceToBatchNdOperator()
|