Home
last modified time | relevance | path

Searched defs:outShape (Results 1 – 24 of 24) sorted by relevance

/external/armnn/src/backends/tosaReference/test/
DTosaRefLayerSupportTests.cpp24 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 …]
/external/armnn/src/backends/reference/workloads/
DElementwiseFunction.cpp25 const TensorShape& outShape, in ElementwiseBinaryFunction()
35 const TensorShape& outShape, in ElementwiseUnaryFunction()
45 const TensorShape& outShape, in LogicalBinaryFunction()
55 const TensorShape& outShape, in LogicalUnaryFunction()
DBroadcast.cpp11 …roadcastLoop(const TensorShape& inShape0, const TensorShape& inShape1, const TensorShape& outShape) in BroadcastLoop()
33 BroadcastLoop::BroadcastLoop(const TensorShape& inShape, const TensorShape& outShape) in BroadcastLoop()
DRefLogicalUnaryWorkload.cpp44 const TensorShape& outShape = outputInfo.GetShape(); in Execute() local
DRefLogicalBinaryWorkload.cpp46 const TensorShape& outShape = outputInfo.GetShape(); in Execute() local
DRefElementwiseUnaryWorkload.cpp54 const TensorShape& outShape = outputInfo.GetShape(); in Execute() local
DRefElementwiseBinaryWorkload.cpp35 const TensorShape& outShape = outputInfo.GetShape(); in ExecuteFunction() local
DRefComparisonWorkload.cpp68 const TensorShape& outShape = outputInfo.GetShape(); in Execute() local
DRefElementwiseWorkload.cpp53 const TensorShape& outShape = outputInfo.GetShape(); in Execute() local
/external/pytorch/test/cpp/tensorexpr/
Dtest_ops.cpp30 const auto& outShape = outputShapes[idx]; in TEST() local
61 const auto& outShape = outputShapes[idx]; in TEST() local
/external/armnn/src/armnn/layers/
DConstantLayer.cpp49 TensorShape const& outShape = m_LayerOutput->GetTensorInfo().GetShape(); in ValidateTensorShapesFromInputs() local
/external/armnn/src/armnn/optimizations/
DPermuteAsReshape.hpp41 const TensorShape& outShape = layer.GetOutputHandler().GetTensorInfo().GetShape(); in IsReshape() local
DTransposeAsReshape.hpp41 const TensorShape& outShape = layer.GetOutputHandler().GetTensorInfo().GetShape(); in IsReshape() local
/external/armnn/src/backends/reference/test/
DRefLayerSupportTests.cpp42 armnn::TensorShape outShape = {1,1,3,4}; variable
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Drestrict_max_rank.cc133 SmallVector<int64_t, 4> outShape; in matchAndRewrite() local
/external/armnn/src/backends/backendsCommon/
DLayerSupportRules.hpp167 const TensorShape& outShape = out.GetShape(); in ShapesAreBroadcastCompatible() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
Dchlo_ops.cc119 llvm::SmallVector<int64_t, 4> outShape; in getBroadcastType() local
150 llvm::SmallVector<int64_t, 4> outShape(shapeLarge.begin(), shapeLarge.end()); in getBroadcastType() local
Dhlo_ops.cc3781 auto outShape = llvm::to_vector<6>(firstType.getShape()); in inferReturnTypes() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
DChloOps.cpp92 llvm::SmallVector<int64_t, 4> outShape; in getBroadcastType() local
123 llvm::SmallVector<int64_t, 4> outShape(shapeLarge.begin(), shapeLarge.end()); in getBroadcastType() local
DStablehloOps.cpp2955 auto outShape = llvm::to_vector<6>(firstType.getShape()); in inferReturnTypes() local
/external/armnn/src/backends/backendsCommon/test/layerTests/
DComparisonTestImpl.cpp40 const armnn::TensorShape& outShape, in ComparisonTestImpl()
103 const armnn::TensorShape outShape, in ComparisonTestImpl()
/external/android-nn-driver/
DConversionUtils.hpp3739 armnn::TensorShape outShape = armnn::TensorShape(outputDims.size(), outputDims.data()); in ConvertSqueeze() local
/external/armnn/shim/sl/canonical/
DConverter.cpp5692 armnn::TensorShape outShape = armnn::TensorShape(outputDims.size(), outputDims.data()); in ConvertSqueeze() local
/external/armnn/src/armnnTfLiteParser/
DTfLiteParser.cpp2220 TensorShape outShape = TensorShape(static_cast<unsigned int>(outputDims.size()), in OutputShapeOfSqueeze() local