Home
last modified time | relevance | path

Searched refs:channelDim (Results 1 – 25 of 55) sorted by relevance

123

/frameworks/ml/nn/runtime/test/specs/V1_2/
Ddepthwise_conv2d_v1_2.mod.py35 …f1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=3, scales=[0.01,…
36 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.005, 0.0025, 0.005,…
41 …f1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=3, scales=[0.01,…
42 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.005, 0.0025, 0.005,…
74 …f2: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=3, scales=[0.5, …
75 …b2: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25, 0.125, 0.25, 0.…
102 …f3: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=3, scales=[0.125…
103 …b3: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.0625, 0.125], hide=…
130 …f4: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=3, scales=[1.0, …
131 …b4: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5, 1.0, 0.5, 0.5], …
Dgrouped_conv2d.mod.py45 …w1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25,…
46 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.0625, 0.125], hide=…
52 …w1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25,…
53 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.0625, 0.125], hide=…
86 …w2: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[2.0, …
87 …b2: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5, 0.625], hide=Tru…
121 …w3: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25,…
122 …b3: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.125, 0.15] * 3, hid…
Dconv2d_v1_2.mod.py35 …f1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.125…
36 … b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.0625], hide=True)),
63 …f2: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5])…
64 b2: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25], hide=True)),
91 …f3: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5, …
92 …b3: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25, 0.2, 0.15], hid…
119 …f4: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5, …
120 …b4: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25, 0.5, 0.25], hid…
125 …f4: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5, …
126 …b4: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5, 1.0, 1.005], hid…
Ddepthwise_conv2d_per_channel.mod.py20 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]))
Dtranspose_conv2d.mod.py46 …w1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25,…
47 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.0625, 0.125], hide=…
53 …w1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25,…
54 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.0625, 0.125], hide=…
87 …w2: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25]…
88 b2: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5], hide=True)),
Dconv2d_per_channel.mod.py19 …[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]))
Ddepthwise_conv2d_invalid_filter_dims.mod.py38 SymmPerChannelQuantParams(channelDim=3, scales=[0.01, 0.005, 0.01, 0.005])),
40 SymmPerChannelQuantParams(channelDim=0, scales=[0.005, 0.0025, 0.005, 0.0025], hide=True)),
Dtranspose_conv2d_large.mod.py39 …w1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5] …
40 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.125] * 16, hide=Tru…
Ddequantize_v1_2.mod.py64 extraParams=SymmPerChannelQuantParams(channelDim=0, scales=[2., 0.5])),
81 channelDim=1, scales=[2., 1., 0.5])),
/frameworks/ml/nn/runtime/test/specs/V1_3/
Dgrouped_conv2d_quant8_signed.mod.py45 …w1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25,…
46 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.0625, 0.125], hide=…
52 …w1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25,…
53 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.0625, 0.125], hide=…
86 …w2: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[2.0, …
87 …b2: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5, 0.625], hide=Tru…
121 …w3: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25,…
122 …b3: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.125, 0.15] * 3, hid…
Ddepthwise_conv2d_quant8_signed.mod.py167 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]))
389 …f1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=3, scales=[0.01,…
390 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.005, 0.0025, 0.005,…
395 …f1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=3, scales=[0.01,…
396 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.005, 0.0025, 0.005,…
429 …f2: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=3, scales=[0.5, …
430 …b2: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25, 0.125, 0.25, 0.…
458 …f3: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=3, scales=[0.125…
[all …]
Dtranspose_conv2d_quant8_signed.mod.py39 …w1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5] …
40 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.125] * 16, hide=Tru…
80 …w1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25,…
81 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.0625, 0.125], hide=…
87 …w1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25,…
88 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.0625, 0.125], hide=…
122 …w2: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25]…
123 b2: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5], hide=True)),
Dconv2d_quant8_signed.mod.py151 …[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]))
225 …f1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.125…
226 … b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.0625], hide=True)),
253 …f2: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5])…
254 b2: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25], hide=True)),
281 …f3: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5, …
282 …b3: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25, 0.2, 0.15], hid…
309 …f4: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5, …
[all …]
Dtranspose_quant8_signed.mod.py39 …w1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5] …
40 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.125] * 16, hide=Tru…
80 …w1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25,…
81 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.0625, 0.125], hide=…
87 …w1: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25,…
88 …b1: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.0625, 0.125], hide=…
122 …w2: ("TENSOR_QUANT8_SYMM_PER_CHANNEL", 0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.25]…
123 b2: ("TENSOR_INT32", 0.0, 0, SymmPerChannelQuantParams(channelDim=0, scales=[0.5], hide=True)),
/frameworks/ml/nn/runtime/test/
DTestOperandExtraParams.cpp109 .channelDim = 3, in createSymmPerChannelQuantParams()
181 .channelDim = 7, in TEST_F()
194 .channelDim = 3, in TEST_F()
199 .channelDim = 3, in TEST_F()
216 .channelDim = 3, in TEST_F()
228 .channelDim = 3, in TEST_F()
/frameworks/ml/nn/runtime/test/specs/V1_3_cts_only/
Dconv2d_v1_3_invalid_rank.mod.py36 SymmPerChannelQuantParams(channelDim=0, scales=[0.125])),
38 SymmPerChannelQuantParams(channelDim=0, scales=[0.0625], hide=True)),
/frameworks/ml/nn/common/operations/
DDequantize.cpp55 const int channelDim = inputShape.extraParams.channelQuant().channelDim; in computePerChannel() local
57 for (int i = getNumberOfDimensions(inputShape) - 1; i > channelDim; --i) { in computePerChannel()
69 const int scaleIndex = (i / stride) % getSizeOfDimension(inputShape, channelDim); in computePerChannel()
/frameworks/ml/nn/runtime/test/android_fuzzing/
DConverter.cpp55 const uint32_t channelDim = params.channel_dim(); in convert() local
56 return {.scales = std::move(scales), .channelDim = channelDim}; in convert()
DFuzzTest.cpp37 SymmPerChannelQuantParams(op.channelQuant.scales, op.channelQuant.channelDim)); in getOperandType()
DDriverFuzzTest.cpp97 .scales = op.channelQuant.scales, .channelDim = op.channelQuant.channelDim}); in createSubgraph()
/frameworks/ml/nn/runtime/test/generated/spec_V1_2/
Ddepthwise_conv2d_per_channel.example.cpp28 .channelDim = 3, in get_test_model_same()
178 .channelDim = 3, in get_test_model_same_all_inputs_as_internal()
362 .channelDim = 3, in get_test_model_same_all_tensors_as_inputs()
512 .channelDim = 3, in get_test_model_same_all_tensors_as_inputs_all_inputs_as_internal()
696 .channelDim = 3, in get_test_model_different()
846 .channelDim = 3, in get_test_model_different_all_inputs_as_internal()
1030 .channelDim = 3, in get_test_model_different_all_tensors_as_inputs()
1180 .channelDim = 3, in get_test_model_different_all_tensors_as_inputs_all_inputs_as_internal()
1364 .channelDim = 3, in get_test_model_layout_nhwc()
1524 .channelDim = 3, in get_test_model_layout_nhwc_all_inputs_as_internal()
[all …]
Dconv2d_per_channel.example.cpp28 .channelDim = 0, in get_test_model()
168 .channelDim = 0, in get_test_model_all_inputs_as_internal()
342 .channelDim = 0, in get_test_model_all_tensors_as_inputs()
482 .channelDim = 0, in get_test_model_all_tensors_as_inputs_all_inputs_as_internal()
656 .channelDim = 0, in get_test_model_layouts_nhwc()
806 .channelDim = 0, in get_test_model_layouts_nhwc_all_inputs_as_internal()
990 .channelDim = 0, in get_test_model_layouts_nhwc_all_tensors_as_inputs()
1140 .channelDim = 0, in get_test_model_layouts_nhwc_all_tensors_as_inputs_all_inputs_as_internal()
1324 .channelDim = 0, in get_test_model_layouts_nchw()
1474 .channelDim = 0, in get_test_model_layouts_nchw_all_inputs_as_internal()
[all …]
Dtranspose_conv2d_large.example.cpp638 .channelDim = 0, in get_test_model_channelQuant8()
768 .channelDim = 0, in get_test_model_channelQuant8_all_inputs_as_internal()
932 .channelDim = 0, in get_test_model_channelQuant8_all_tensors_as_inputs()
1062 .channelDim = 0, in get_test_model_channelQuant8_all_tensors_as_inputs_all_inputs_as_internal()
/frameworks/ml/nn/runtime/include/
DNeuralNetworksWrapper.h86 SymmPerChannelQuantParams(std::vector<float> scalesVec, uint32_t channelDim) in SymmPerChannelQuantParams()
89 .channelDim = channelDim, in SymmPerChannelQuantParams()
/frameworks/ml/nn/tools/test_generator/
Dexample_generator.py100 return {"scales": extraParams.scales, "channelDim": extraParams.channelDim}

123