Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/passes/onnx/
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 …]
/external/armnn/src/backends/tosaCommon/test/
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
/external/armnn/src/armnnSerializer/test/
DSerializerTestUtils.cpp78 void LayerVerifierBase::VerifyConstTensors(const std::string& tensorName, in VerifyConstTensors()
/external/pytorch/torch/csrc/jit/serialization/
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
/external/armnn/src/backends/backendsCommon/
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()
/external/armnn/shim/sl/canonical/
DCanonicalUtils.cpp219 const std::string& tensorName, in DumpTensor()
/external/android-nn-driver/
DUtils.cpp339 const std::string& tensorName, in DumpTensor()
/external/armnn/src/backends/backendsCommon/test/layerTests/
DTransposeConvolution2dTestImpl.cpp36 void VerifyInputTensorData(const TensorData<T>& data, const std::string& tensorName) in VerifyInputTensorData()
/external/armnn/src/armnnOnnxParser/
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()
/external/armnn/src/backends/reference/
DRefLayerSupport.cpp51 std::string& tensorName) in CreateIncorrectDimensionsErrorMsg()