Home
last modified time | relevance | path

Searched defs:tensorName (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dconstant_map.cpp19 const std::string& tensorName, in SetRank()
25 bool ConstantValueMap::HasRank(const std::string& tensorName) { in HasRank()
30 std::optional<size_t> ConstantValueMap::GetRank(const std::string& tensorName) { in GetRank()
55 const std::string& tensorName, in SetShape()
61 bool ConstantValueMap::HasShape(const std::string& tensorName) { in HasShape()
67 const std::string& tensorName) { in GetShape()
75 const std::string& tensorName, in SetValue()
80 bool ConstantValueMap::HasValue(const std::string& tensorName) { in HasValue()
86 const std::string& tensorName) { in GetValue()
93 void ConstantValueMap::EraseValue(const std::string& tensorName) { in EraseValue()
[all …]
/aosp_15_r20/external/armnn/src/backends/tosaCommon/test/
H A DTosaTestUtils.hpp210 std::basic_string<char> tensorName = basicBlock->GetTensors()[i]->GetName(); in AssertTosaOneToOneMappingBasicBlock() local
223 std::basic_string<char> tensorName = basicBlock->GetTensors()[numInputs + i]->GetName(); in AssertTosaOneToOneMappingBasicBlock() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dexport.cpp193 auto tensorName = external_ref.value(); in GetExternalFileName() local
209 const std::string& tensorName, in CreateExternalFile()
1324 auto tensorName = GetExternalFileName(tensor_proto->name()); in EncodeTensor() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadData.cpp113 std::string const& tensorName) in ValidateTensorNumElements()
125 const std::string& descName, std::string const& tensorName) in ValidateTensorDataType()
134 …tizedDataType(const TensorInfo& tensor, const std::string& descName, const std::string& tensorName) in ValidPerAxisQuantizedDataType()
355 const std::string& tensorName) in ValidatePerAxisQuantizationDimension()
367 const std::string& tensorName) in ValidatePerAxisQuantizationOffset()
/aosp_15_r20/external/armnn/src/armnnSerializer/test/
H A DSerializerTestUtils.cpp78 void LayerVerifierBase::VerifyConstTensors(const std::string& tensorName, in VerifyConstTensors()
/aosp_15_r20/external/armnn/src/armnnOnnxParser/
H A DOnnxParser.cpp109 std::string tensorName, in CheckValidDataType()
1425 void OnnxParserImpl::CreateConstantLayer(const std::string& tensorName, const std::string& layerNam… in CreateConstantLayer()
1433 void OnnxParserImpl::CreateInt64ConstantLayer(const std::string& tensorName, const std::string& lay… in CreateInt64ConstantLayer()
/aosp_15_r20/external/armnn/shim/sl/canonical/
H A DCanonicalUtils.cpp219 const std::string& tensorName, in DumpTensor()
/aosp_15_r20/external/android-nn-driver/
H A DUtils.cpp339 const std::string& tensorName, in DumpTensor()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DTransposeConvolution2dTestImpl.cpp36 void VerifyInputTensorData(const TensorData<T>& data, const std::string& tensorName) in VerifyInputTensorData()
/aosp_15_r20/external/armnn/src/backends/reference/
H A DRefLayerSupport.cpp51 std::string& tensorName) in CreateIncorrectDimensionsErrorMsg()