| /aosp_15_r20/external/armnn/src/backends/tosaReference/test/ |
| H A D | TosaRefLayerSupportTests.cpp | 24 TensorShape outShape = {1,1,3,4}; variable
46 TensorShape outShape = {1,1,3,4}; variable
197 TensorShape outShape = {1,1,3,4}; variable
219 TensorShape outShape = {1,1,3,4}; variable
240 TensorShape outShape = {1,1,3,4}; variable
266 TensorShape outShape = {1,1,3,4}; variable
293 TensorShape outShape = {1,1,3,4}; variable
313 TensorShape outShape = {1,1,3,4}; variable
336 TensorShape outShape = {12}; variable
358 TensorShape outShape = {12}; variable
[all …]
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 486 Shape outShape = from.shape(); in convertFromNhwc() local
514 Shape outShape = from.shape(); in convertFromNhwc() local
803 Shape outShape = output.shape(); in executeOperation() local
821 Shape outShape = output.shape(); in executeOperation() local
885 Shape outShape = output.shape(); in executeOperation() local
1106 Shape outShape = output.shape(); in executeOperation() local
1178 Shape outShape = output.shape(); in executeOperation() local
1254 Shape outShape = output.shape(); in executeOperation() local
1293 Shape outShape = output.shape(); in executeOperation() local
1308 Shape outShape = output.shape(); in executeOperation() local
[all …]
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | ElementwiseFunction.cpp | 25 const TensorShape& outShape, in ElementwiseBinaryFunction()
35 const TensorShape& outShape, in ElementwiseUnaryFunction()
45 const TensorShape& outShape, in LogicalBinaryFunction()
55 const TensorShape& outShape, in LogicalUnaryFunction()
|
| H A D | Broadcast.cpp | 11 …roadcastLoop(const TensorShape& inShape0, const TensorShape& inShape1, const TensorShape& outShape) in BroadcastLoop()
33 BroadcastLoop::BroadcastLoop(const TensorShape& inShape, const TensorShape& outShape) in BroadcastLoop()
|
| H A D | RefLogicalUnaryWorkload.cpp | 44 const TensorShape& outShape = outputInfo.GetShape(); in Execute() local
|
| H A D | RefLogicalBinaryWorkload.cpp | 46 const TensorShape& outShape = outputInfo.GetShape(); in Execute() local
|
| H A D | RefElementwiseUnaryWorkload.cpp | 54 const TensorShape& outShape = outputInfo.GetShape(); in Execute() local
|
| H A D | RefElementwiseBinaryWorkload.cpp | 35 const TensorShape& outShape = outputInfo.GetShape(); in ExecuteFunction() local
|
| H A D | RefComparisonWorkload.cpp | 68 const TensorShape& outShape = outputInfo.GetShape(); in Execute() local
|
| H A D | RefElementwiseWorkload.cpp | 53 const TensorShape& outShape = outputInfo.GetShape(); in Execute() local
|
| /aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
| H A D | test_ops.cpp | 30 const auto& outShape = outputShapes[idx]; in TEST() local
61 const auto& outShape = outputShapes[idx]; in TEST() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Elementwise.cpp | 125 const Shape outShape = context->getOutputShape(kOutputTensor); in executeRsqrt() local
135 const Shape outShape = context->getOutputShape(kOutputTensor); in executeRsqrt() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| H A D | ChloOps.cpp | 92 llvm::SmallVector<int64_t, 4> outShape; in getBroadcastType() local
123 llvm::SmallVector<int64_t, 4> outShape(shapeLarge.begin(), shapeLarge.end()); in getBroadcastType() local
|
| H A D | StablehloOps.cpp | 2955 auto outShape = llvm::to_vector<6>(firstType.getShape()); in inferReturnTypes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/ |
| H A D | chlo_ops.cc | 119 llvm::SmallVector<int64_t, 4> outShape; in getBroadcastType() local
150 llvm::SmallVector<int64_t, 4> outShape(shapeLarge.begin(), shapeLarge.end()); in getBroadcastType() local
|
| /aosp_15_r20/external/armnn/src/armnn/layers/ |
| H A D | ConstantLayer.cpp | 49 TensorShape const& outShape = m_LayerOutput->GetTensorInfo().GetShape(); in ValidateTensorShapesFromInputs() local
|
| /aosp_15_r20/external/armnn/src/armnn/optimizations/ |
| H A D | PermuteAsReshape.hpp | 41 const TensorShape& outShape = layer.GetOutputHandler().GetTensorInfo().GetShape(); in IsReshape() local
|
| H A D | TransposeAsReshape.hpp | 41 const TensorShape& outShape = layer.GetOutputHandler().GetTensorInfo().GetShape(); in IsReshape() local
|
| /aosp_15_r20/external/armnn/src/backends/reference/test/ |
| H A D | RefLayerSupportTests.cpp | 42 armnn::TensorShape outShape = {1,1,3,4}; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| H A D | restrict_max_rank.cc | 133 SmallVector<int64_t, 4> outShape; in matchAndRewrite() local
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
| H A D | ComparisonTestImpl.cpp | 40 const armnn::TensorShape& outShape, in ComparisonTestImpl()
103 const armnn::TensorShape outShape, in ComparisonTestImpl()
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/ |
| H A D | LayerSupportRules.hpp | 167 const TensorShape& outShape = out.GetShape(); in ShapesAreBroadcastCompatible() local
|
| /aosp_15_r20/external/android-nn-driver/ |
| H A D | ConversionUtils.hpp | 3739 armnn::TensorShape outShape = armnn::TensorShape(outputDims.size(), outputDims.data()); in ConvertSqueeze() local
|
| /aosp_15_r20/external/armnn/shim/sl/canonical/ |
| H A D | Converter.cpp | 5692 armnn::TensorShape outShape = armnn::TensorShape(outputDims.size(), outputDims.data()); in ConvertSqueeze() local
|
| /aosp_15_r20/external/armnn/src/armnnTfLiteParser/ |
| H A D | TfLiteParser.cpp | 2220 TensorShape outShape = TensorShape(static_cast<unsigned int>(outputDims.size()), in OutputShapeOfSqueeze() local
|