| /aosp_15_r20/external/armnn/include/armnnTestUtils/ |
| H A D | MemCopyTestImpl.hpp | 24 const armnn::TensorShape tensorShape(4, shapeData.data()); in MemCopyTest() local
|
| /aosp_15_r20/external/armnn/src/armnn/layers/ |
| H A D | TransposeConvolution2dLayer.cpp | 83 TensorShape tensorShape = m_Param.m_DataLayout == armnn::DataLayout::NHWC ? in InferOutputShapes() local
|
| H A D | Pooling2dLayer.cpp | 98 TensorShape tensorShape = m_Param.m_DataLayout == armnn::DataLayout::NHWC ? in InferOutputShapes() local
|
| H A D | Pooling3dLayer.cpp | 102 TensorShape tensorShape = m_Param.m_DataLayout == armnn::DataLayout::NDHWC ? in InferOutputShapes() local
|
| H A D | Convolution2dLayer.cpp | 95 TensorShape tensorShape = m_Param.m_DataLayout == armnn::DataLayout::NHWC ? in InferOutputShapes() local
|
| H A D | Convolution3dLayer.cpp | 100 TensorShape tensorShape = m_Param.m_DataLayout == armnn::DataLayout::NDHWC ? in InferOutputShapes() local
|
| /aosp_15_r20/external/armnn/delegate/test/ |
| H A D | RoundTestHelper.hpp | 23 std::vector<char> CreateRoundTfLiteModel(tflite::BuiltinOperator roundOperatorCode, in CreateRoundTfLiteModel()
|
| H A D | CastTestHelper.hpp | 23 std::vector<char> CreateCastTfLiteModel(tflite::TensorType inputTensorType, in CreateCastTfLiteModel()
|
| H A D | SoftmaxTestHelper.hpp | 24 std::vector<char> CreateSoftmaxTfLiteModel(tflite::BuiltinOperator softmaxOperatorCode, in CreateSoftmaxTfLiteModel()
|
| H A D | ElementwiseUnaryTestHelper.hpp | 24 std::vector<char> CreateElementwiseUnaryTfLiteModel(tflite::BuiltinOperator unaryOperatorCode, in CreateElementwiseUnaryTfLiteModel()
|
| /aosp_15_r20/external/armnn/samples/common/include/ArmnnUtils/ |
| H A D | ArmnnNetworkExecutor.hpp | 142 const armnn::TensorShape& tensorShape = m_outputBindingInfo[it].second.GetShape(); in ArmnnNetworkExecutor() local
|
| /aosp_15_r20/external/armnn/src/armnnTfLiteParser/test/ |
| H A D | Unsupported.cpp | 181 static std::string GetTensorShapeAsString(const TensorShape& tensorShape) in GetTensorShapeAsString()
|
| /aosp_15_r20/external/armnn/delegate/classic/src/ |
| H A D | ClassicDelegateUtils.hpp | 385 armnn::TensorShape tensorShape(armnn::Dimensionality::NotSpecified); in GetTensorInfoForTfLiteTensor() local
|
| /aosp_15_r20/external/armnn/delegate/opaque/src/ |
| H A D | OpaqueDelegateUtils.hpp | 461 armnn::TensorShape tensorShape(armnn::Dimensionality::NotSpecified); in GetTensorInfoForTfLiteOpaqueTensor() local
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
| H A D | BatchNormalizationTestImpl.cpp | 630 constexpr unsigned int tensorShape[] = {channels}; in CompareBatchNormTest() local
|
| /aosp_15_r20/external/pytorch/android/pytorch_android/src/main/cpp/ |
| H A D | pytorch_jni_common.cpp | 230 const auto& tensorShape = tensor.sizes(); in newJTensorFromAtTensor() local
|
| /aosp_15_r20/external/armnn/src/armnnTestUtils/ |
| H A D | CreateWorkload.hpp | 372 TensorShape tensorShape; in CreateBatchNormalizationWorkloadTest() local
430 TensorShape tensorShape; in CreateBatchNormalizationWithBlobWorkloadTest() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | ProcessGroupGloo.cpp | 1094 const c10::IntArrayRef tensorShape = tensor.sizes(); in allreduceCoalesced() local
1366 auto tensorShape = std::vector<int64_t>({(int64_t)nnz}); in allgather_values() local
|
| /aosp_15_r20/external/armnn/src/armnnTfLiteParser/ |
| H A D | TfLiteParser.cpp | 467 TensorShape tensorShape; in ToTensorInfo() local
|