Searched defs:tensorName (Results 1 – 10 of 10) sorted by relevance
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
H A D | constant_map.cpp | 19 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 D | TosaTestUtils.hpp | 210 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 D | export.cpp | 193 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 D | WorkloadData.cpp | 113 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 D | SerializerTestUtils.cpp | 78 void LayerVerifierBase::VerifyConstTensors(const std::string& tensorName, in VerifyConstTensors()
|
/aosp_15_r20/external/armnn/src/armnnOnnxParser/ |
H A D | OnnxParser.cpp | 109 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 D | CanonicalUtils.cpp | 219 const std::string& tensorName, in DumpTensor()
|
/aosp_15_r20/external/android-nn-driver/ |
H A D | Utils.cpp | 339 const std::string& tensorName, in DumpTensor()
|
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
H A D | TransposeConvolution2dTestImpl.cpp | 36 void VerifyInputTensorData(const TensorData<T>& data, const std::string& tensorName) in VerifyInputTensorData()
|
/aosp_15_r20/external/armnn/src/backends/reference/ |
H A D | RefLayerSupport.cpp | 51 std::string& tensorName) in CreateIncorrectDimensionsErrorMsg()
|