| /packages/modules/NeuralNetworks/runtime/include/ |
| D | NeuralNetworksWrapperExtensions.h | 54 ExtraParams extraParams; member
59 extraParams(other.extraParams) { in OperandType()
67 extraParams = other.extraParams;
74 ExtraParams&& extraParams = std::monostate())
75 : dimensions(std::move(d)), extraParams(std::move(extraParams)) { in dimensions()
89 OperandType(Type type, std::vector<uint32_t> dimensions, ExtraParams&& extraParams) in OperandType()
90 : OperandType(type, dimensions, 0.0f, 0, std::move(extraParams)) {} in OperandType()
122 if (std::holds_alternative<SymmPerChannelQuantParams>(type->extraParams)) { in addOperand()
123 const auto& channelQuant = std::get<SymmPerChannelQuantParams>(type->extraParams); in addOperand()
128 } else if (std::holds_alternative<ExtensionOperandParams>(type->extraParams)) { in addOperand()
[all …]
|
| /packages/modules/NeuralNetworks/tools/test_generator/ |
| D | test_generator.py | 55 if isinstance(ty.extraParams, SymmPerChannelQuantParams):
56 v *= ty.extraParams.GetScalesBroadcastArray(ty.dimensions)
63 if isinstance(ty.extraParams, SymmPerChannelQuantParams):
64 v = v / ty.extraParams.GetScalesBroadcastArray(ty.dimensions)
159 extraParams=None): argument
165 self.extraParams = extraParams
170 def GetType(vt, dimensions, scale=0, zeroPoint=0, extraParams=None): argument
173 key = ",".join([vt, str(dimensions), str(scale), str(zeroPoint), str(extraParams)])
175 Type.typesMap[key] = Type(vt, dimensions, scale, zeroPoint, extraParams=extraParams)
185 def GetTypeFromString(vt, shape, extraParams=None): argument
[all …]
|
| D | example_generator.py | 95 def GetSymmPerChannelQuantParams(extraParams): argument
97 if extraParams is None or extraParams.hide:
100 return {"scales": extraParams.scales, "channelDim": extraParams.channelDim}
112 "channelQuant": GetSymmPerChannelQuantParams(operand.type.extraParams),
|
| D | spec_visualizer.py | 84 op_info["scale"] = FormatArray(op.type.extraParams.scales)
85 op_info["channel dim"] = op.type.extraParams.channelDim
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | depthwise_conv2d_per_channel.mod.py | 20 extraParams = SymmPerChannelQuantParams(channelDim=3, scales=[0.5, 0.5]))
36 extraParams = SymmPerChannelQuantParams(channelDim=3, scales=[1.0, 0.5, 1.0, 0.5]))
55 extraParams = SymmPerChannelQuantParams(channelDim=3, scales=[1.0, 0.5, 1.0, 0.5]))
|
| D | conv2d_per_channel.mod.py | 19 …[1, 2, 1, 2, 1, 2], extraParams = SymmPerChannelQuantParams(channelDim=0, scales=[0.5, 0.75, 1.0]))
34 …[1, 2, 1, 2, 1, 2], extraParams = SymmPerChannelQuantParams(channelDim=0, scales=[0.5, 0.75, 1.0]))
63 …[1, 2, 1, 2, 1, 2], extraParams = SymmPerChannelQuantParams(channelDim=0, scales=[0.5, 0.75, 1.0]))
|
| D | dequantize_v1_2.mod.py | 64 extraParams=SymmPerChannelQuantParams(channelDim=0, scales=[2., 0.5])),
80 extraParams=SymmPerChannelQuantParams(
|
| /packages/modules/NeuralNetworks/common/ |
| D | ValidateHal.cpp | 115 NN_RET_CHECK(operand.extraParams.getDiscriminator() == in validateOperandExtraParams()
119 << " has incorrect extraParams: " << toString(operand.extraParams); in validateOperandExtraParams()
122 NN_RET_CHECK(operand.extraParams.getDiscriminator() == in validateOperandExtraParams()
126 auto& channelQuant = operand.extraParams.channelQuant(); in validateOperandExtraParams()
152 NN_RET_CHECK(operand.extraParams.getDiscriminator() == in validateOperandExtraParams()
154 operand.extraParams.getDiscriminator() == in validateOperandExtraParams()
158 << " has incorrect extraParams: " << toString(operand.extraParams); in validateOperandExtraParams()
923 NN_RET_CHECK(operand.extraParams == operands[0].extraParams) in validateMemoryDesc()
924 << toString(operand.extraParams) << " vs " << toString(operands[0].extraParams); in validateMemoryDesc()
|
| D | TypeUtils.cpp | 698 std::ostream& operator<<(std::ostream& os, const Operand::ExtraParams& extraParams) { in operator <<() argument
700 if (std::holds_alternative<Operand::NoParams>(extraParams)) { in operator <<()
702 } else if (std::holds_alternative<Operand::SymmPerChannelQuantParams>(extraParams)) { in operator <<()
703 os << std::get<Operand::SymmPerChannelQuantParams>(extraParams); in operator <<()
704 } else if (std::holds_alternative<Operand::ExtensionParams>(extraParams)) { in operator <<()
705 os << std::get<Operand::ExtensionParams>(extraParams); in operator <<()
714 << ", .extraParams=" << operand.extraParams << "}"; in operator <<()
993 operand.lifetime, operand.location, operand.extraParams); in operator ==()
|
| D | AidlValidateHal.cpp | 97 const auto& lhsExtraParams = operand.extraParams; in validateMemoryDesc()
98 const auto& rhsExtraParams = operands[0].extraParams; in validateMemoryDesc()
|
| D | Validation.cpp | 615 NN_VALIDATE(std::holds_alternative<Operand::NoParams>(operand.extraParams)) in validateOperandExtraParams()
621 std::holds_alternative<Operand::SymmPerChannelQuantParams>(operand.extraParams)) in validateOperandExtraParams()
625 std::get<Operand::SymmPerChannelQuantParams>(operand.extraParams); in validateOperandExtraParams()
653 NN_VALIDATE(std::holds_alternative<Operand::NoParams>(operand.extraParams) || in validateOperandExtraParams()
654 std::holds_alternative<Operand::ExtensionParams>(operand.extraParams)) in validateOperandExtraParams()
1317 NN_VALIDATE_EQ(operand.extraParams, operands.front().extraParams) in validateMemoryDescImpl()
1318 << operand.extraParams << " vs " << operands.front().extraParams; in validateMemoryDescImpl()
1403 operand->extraParams}; in getInputShape()
1407 return getInputOperand(index)->extraParams; in getInputExtraParams()
1417 operand->extraParams}; in getOutputShape()
[all …]
|
| D | CpuExecutor.cpp | 135 return getInputInfo(index)->extraParams; in getInputExtraParams()
187 if (info->extraParams != shape.extraParams) { in setInfoAndAllocateIfNeeded()
204 info->extraParams = shape.extraParams; in setInfoAndAllocateIfNeeded()
625 to.extraParams = from.extraParams; in initializeRunTimeInfo()
1551 std::get<Operand::SymmPerChannelQuantParams>(filter.extraParams) in executeOperation()
1571 std::get<Operand::SymmPerChannelQuantParams>(filter.extraParams) in executeOperation()
|
| D | LegacyUtils.cpp | 243 operand->extraParams}; in getInputShape()
247 return getInputOperand(index)->extraParams; in getInputExtraParams()
257 operand->extraParams}; in getOutputShape()
597 NN_RET_CHECK(a.extraParams == b.extraParams) << a.extraParams << " != " << b.extraParams; in compatible()
1607 operands[inputIndexes[1]].extraParams) in validateOperation()
|
| D | MetaModel.cpp | 203 return compare(a.extraParams, b.extraParams); in operator ()()
|
| D | LegacyHalUtils.cpp | 771 if (operand.extraParams.getDiscriminator() == in compliantWithAidl()
773 operand.extraParams.channelQuant().channelDim > std::numeric_limits<int32_t>::max()) { in compliantWithAidl()
1180 .extraParams = operand.extraParams}; in convertToV1_2()
1201 .extraParams = operand.extraParams}; in convertToV1_3()
|
| /packages/modules/NeuralNetworks/tools/test_generator/test_harness/ |
| D | TestUtils.cpp | 69 Operand::ExtraParams extraParams; in createOperand() local
71 extraParams = in createOperand()
83 .extraParams = std::move(extraParams), in createOperand()
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | CpuExecutor.h | 74 Operand::ExtraParams extraParams; member
82 .extraParams = extraParams, in shape()
|
| /packages/modules/NeuralNetworks/runtime/test/android_fuzzing/ |
| D | DriverFuzzTest.cpp | 94 V1_2::Operand::ExtraParams extraParams; in createSubgraph() local
96 extraParams.channelQuant(V1_2::SymmPerChannelQuantParams{ in createSubgraph()
107 .extraParams = std::move(extraParams)}; in createSubgraph()
|
| /packages/modules/NeuralNetworks/runtime/ |
| D | ModelBuilder.cpp | 104 .extraParams = {}, in addOperand()
233 Operand::SymmPerChannelQuantParams extraParams = { in setOperandSymmPerChannelQuantParams() local
238 operand, extraParams, "ANeuralNetworksModel_setOperandSymmPerChannelQuantParams"); in setOperandSymmPerChannelQuantParams()
245 operand.extraParams = std::move(extraParams); in setOperandSymmPerChannelQuantParams()
276 operand.extraParams = {}; in setOperandExtensionData()
278 operand.extraParams = Operand::ExtensionParams( in setOperandExtensionData()
|
| D | Memory.cpp | 144 metadata.operand->extraParams == kOperand.extraParams)); in updateMetadata()
354 operand.extraParams != mOperand->extraParams) { in addRole()
412 LOG(INFO) << " Extra params: " << operand.extraParams; in logMemoryDescriptorToInfo()
|
| D | ModelBuilder.h | 51 uint32_t index, const ANeuralNetworksSymmPerChannelQuantParams& extraParams);
|
| /packages/modules/NeuralNetworks/shim_and_sl/ |
| D | ShimConverter.cpp | 83 const auto& params = operand.extraParams->get<OperandExtraParams::Tag::channelQuant>(); in convertSubgraphFromHAL()
116 if (operand.extraParams && in convertSubgraphFromHAL()
117 operand.extraParams->getTag() == OperandExtraParams::Tag::extension) { in convertSubgraphFromHAL()
119 operand.extraParams->get<OperandExtraParams::Tag::extension>(); in convertSubgraphFromHAL()
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | depthwise_conv2d_quant8_signed.mod.py | 167 extraParams = SymmPerChannelQuantParams(channelDim=3, scales=[0.5, 0.5]))
184 extraParams = SymmPerChannelQuantParams(channelDim=3, scales=[1.0, 0.5, 1.0, 0.5]))
204 extraParams = SymmPerChannelQuantParams(channelDim=3, scales=[1.0, 0.5, 1.0, 0.5]))
|
| D | conv2d_quant8_signed.mod.py | 151 …[1, 2, 1, 2, 1, 2], extraParams = SymmPerChannelQuantParams(channelDim=0, scales=[0.5, 0.75, 1.0]))
166 …[1, 2, 1, 2, 1, 2], extraParams = SymmPerChannelQuantParams(channelDim=0, scales=[0.5, 0.75, 1.0]))
195 …[1, 2, 1, 2, 1, 2], extraParams = SymmPerChannelQuantParams(channelDim=0, scales=[0.5, 0.75, 1.0]))
|
| /packages/modules/NeuralNetworks/common/operations/ |
| D | Dequantize.cpp | 53 std::get<Operand::SymmPerChannelQuantParams>(inputShape.extraParams).channelDim; in computePerChannel()
68 const float scale = std::get<Operand::SymmPerChannelQuantParams>(inputShape.extraParams) in computePerChannel()
|