Home
last modified time | relevance | path

Searched refs:tensorValue (Results 1 – 2 of 2) sorted by relevance

/external/armnn/src/armnn/optimizations/
DFoldPadIntoLayer2d.hpp56 …utralElement(const Convolution2dDescriptor&, const TensorInfo& tensorInfo, const float tensorValue) in IsNeutralElement() argument
58 return tensorValue == GetZeroElement(tensorInfo); in IsNeutralElement()
63 const float tensorValue) in IsNeutralElement() argument
65 return tensorValue == GetZeroElement(tensorInfo); in IsNeutralElement()
69 const Pooling2dDescriptor& descriptor, const TensorInfo& tensorInfo, const float tensorValue) in IsNeutralElement() argument
72 ? tensorValue <= GetLowestElement(tensorInfo) in IsNeutralElement()
73 : tensorValue == GetZeroElement(tensorInfo); in IsNeutralElement()
/external/pytorch/torch/csrc/jit/backends/xnnpack/serialization/
Dserializer.cpp63 const auto tensorValue = CreateXNNTensorValueDirect( in serializeTensorValue() local
74 CreateXValue(_builder, XValueUnion::XNNTensorValue, tensorValue.Union()); in serializeTensorValue()