/frameworks/ml/nn/runtime/test/specs/V1_2/ |
D | depthwise_conv2d_v1_2.mod.py | 35 …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], …
|
D | grouped_conv2d.mod.py | 45 …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…
|
D | conv2d_v1_2.mod.py | 35 …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…
|
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 | transpose_conv2d.mod.py | 46 …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)),
|
D | transpose_conv2d_large.mod.py | 39 …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…
|
D | dequantize_v1_2.mod.py | 64 extraParams=SymmPerChannelQuantParams(channelDim=0, scales=[2., 0.5])), 81 channelDim=1, scales=[2., 1., 0.5])),
|
/frameworks/ml/nn/runtime/test/ |
D | TestOperandExtraParams.cpp | 101 .channelDim = 3, in createSymmPerChannelQuantParams() 171 .channelDim = 7, in TEST_F() 184 .channelDim = 3, in TEST_F() 189 .channelDim = 3, in TEST_F() 206 .channelDim = 3, in TEST_F() 218 .channelDim = 3, in TEST_F()
|
/frameworks/ml/nn/common/operations/ |
D | Dequantize.cpp | 53 const int channelDim = inputShape.extraParams.channelQuant().channelDim; in computePerChannel() local 55 for (int i = getNumberOfDimensions(inputShape) - 1; i > channelDim; --i) { in computePerChannel() 67 const int scaleIndex = (i / stride) % getSizeOfDimension(inputShape, channelDim); in computePerChannel()
|
/frameworks/ml/nn/runtime/test/generated/vts_models/ |
D | depthwise_conv2d_per_channel.model.cpp | 6 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.5f, 0.5f}, .channelDim=3}); in createTestModel_same() 152 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.5f, 0.5f}, .channelDim=3}); in createTestModel_same_weight_as_input() 298 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.5f, 0.5f}, .channelDim=3}); in createTestModel_same_dynamic_output_shape() 444 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.5f, 0.5f}, .channelDim=3}); in createTestModel_same_dynamic_output_shape_weight_as_input() 590 …raParams1.channelQuant(SymmPerChannelQuantParams{.scales={1.0f, 0.5f, 1.0f, 0.5f}, .channelDim=3}); in createTestModel_different() 736 …raParams1.channelQuant(SymmPerChannelQuantParams{.scales={1.0f, 0.5f, 1.0f, 0.5f}, .channelDim=3}); in createTestModel_different_weight_as_input() 882 …raParams1.channelQuant(SymmPerChannelQuantParams{.scales={1.0f, 0.5f, 1.0f, 0.5f}, .channelDim=3}); in createTestModel_different_dynamic_output_shape() 1028 …raParams1.channelQuant(SymmPerChannelQuantParams{.scales={1.0f, 0.5f, 1.0f, 0.5f}, .channelDim=3}); in createTestModel_different_dynamic_output_shape_weight_as_input() 1174 …raParams1.channelQuant(SymmPerChannelQuantParams{.scales={1.0f, 0.5f, 1.0f, 0.5f}, .channelDim=3}); in createTestModel_layout_nhwc() 1329 …raParams1.channelQuant(SymmPerChannelQuantParams{.scales={1.0f, 0.5f, 1.0f, 0.5f}, .channelDim=3}); in createTestModel_layout_nhwc_weight_as_input() [all …]
|
D | conv2d_per_channel.model.cpp | 6 … extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.5f, 0.75f, 1.0f}, .channelDim=0}); in createTestModel() 143 … extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.5f, 0.75f, 1.0f}, .channelDim=0}); in createTestModel_weight_as_input() 280 … extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.5f, 0.75f, 1.0f}, .channelDim=0}); in createTestModel_dynamic_output_shape() 417 … extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.5f, 0.75f, 1.0f}, .channelDim=0}); in createTestModel_dynamic_output_shape_weight_as_input() 554 … extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.5f, 0.75f, 1.0f}, .channelDim=0}); in createTestModel_layouts_nhwc() 700 … extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.5f, 0.75f, 1.0f}, .channelDim=0}); in createTestModel_layouts_nhwc_weight_as_input() 846 … extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.5f, 0.75f, 1.0f}, .channelDim=0}); in createTestModel_layouts_nchw() 992 … extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.5f, 0.75f, 1.0f}, .channelDim=0}); in createTestModel_layouts_nchw_weight_as_input() 1138 … extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.5f, 0.75f, 1.0f}, .channelDim=0}); in createTestModel_layouts_dynamic_output_shape_nhwc() 1284 … extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.5f, 0.75f, 1.0f}, .channelDim=0}); in createTestModel_layouts_dynamic_output_shape_nhwc_weight_as_input() [all …]
|
D | dequantize_v1_2.model.cpp | 1238 extraParams0.channelQuant(SymmPerChannelQuantParams{.scales={2.0f, 0.5f}, .channelDim=0}); in createTestModel_5() 1292 extraParams0.channelQuant(SymmPerChannelQuantParams{.scales={2.0f, 0.5f}, .channelDim=0}); in createTestModel_relaxed_5() 1347 extraParams0.channelQuant(SymmPerChannelQuantParams{.scales={2.0f, 0.5f}, .channelDim=0}); in createTestModel_float16_5() 1401 extraParams0.channelQuant(SymmPerChannelQuantParams{.scales={2.0f, 0.5f}, .channelDim=0}); in createTestModel_dynamic_output_shape_5() 1455 extraParams0.channelQuant(SymmPerChannelQuantParams{.scales={2.0f, 0.5f}, .channelDim=0}); in createTestModel_dynamic_output_shape_relaxed_5() 1510 extraParams0.channelQuant(SymmPerChannelQuantParams{.scales={2.0f, 0.5f}, .channelDim=0}); in createTestModel_dynamic_output_shape_float16_5() 1564 extraParams0.channelQuant(SymmPerChannelQuantParams{.scales={2.0f, 1.0f, 0.5f}, .channelDim=1}); in createTestModel_6() 1618 extraParams0.channelQuant(SymmPerChannelQuantParams{.scales={2.0f, 1.0f, 0.5f}, .channelDim=1}); in createTestModel_relaxed_6() 1673 extraParams0.channelQuant(SymmPerChannelQuantParams{.scales={2.0f, 1.0f, 0.5f}, .channelDim=1}); in createTestModel_float16_6() 1727 extraParams0.channelQuant(SymmPerChannelQuantParams{.scales={2.0f, 1.0f, 0.5f}, .channelDim=1}); in createTestModel_dynamic_output_shape_6() [all …]
|
D | transpose_conv2d_large.model.cpp | 131 …5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f}, .channelDim=0}); in createTestModel_channelQuant8() 384 …5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f, 0.5f}, .channelDim=0}); in createTestModel_dynamic_output_shape_channelQuant8()
|
D | grouped_conv2d.model.cpp | 1224 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_none_channelQuant8() 1379 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_none_channelQuant8_weight_as_input() 1534 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_none_channelQuant8_2() 1689 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_none_channelQuant8_weight_as_input_2() 3366 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_relu_channelQuant8() 3521 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_relu_channelQuant8_weight_as_input() 3676 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_relu_channelQuant8_2() 3831 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_relu_channelQuant8_weight_as_input_2() 5508 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_relu1_channelQuant8() 5663 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_relu1_channelQuant8_weight_as_input() [all …]
|
D | depthwise_conv2d_v1_2.model.cpp | 463 …ms1.channelQuant(SymmPerChannelQuantParams{.scales={0.01f, 0.005f, 0.01f, 0.005f}, .channelDim=3}); in createTestModel_nhwc_channelQuant8() 618 …ms1.channelQuant(SymmPerChannelQuantParams{.scales={0.01f, 0.005f, 0.01f, 0.005f}, .channelDim=3}); in createTestModel_nhwc_channelQuant8_2() 1382 …ms1.channelQuant(SymmPerChannelQuantParams{.scales={0.01f, 0.005f, 0.01f, 0.005f}, .channelDim=3}); in createTestModel_nhwc_weight_as_input_channelQuant8() 1537 …ms1.channelQuant(SymmPerChannelQuantParams{.scales={0.01f, 0.005f, 0.01f, 0.005f}, .channelDim=3}); in createTestModel_nhwc_weight_as_input_channelQuant8_2() 2301 …ms1.channelQuant(SymmPerChannelQuantParams{.scales={0.01f, 0.005f, 0.01f, 0.005f}, .channelDim=3}); in createTestModel_nchw_channelQuant8() 2456 …ms1.channelQuant(SymmPerChannelQuantParams{.scales={0.01f, 0.005f, 0.01f, 0.005f}, .channelDim=3}); in createTestModel_nchw_channelQuant8_2() 3220 …ms1.channelQuant(SymmPerChannelQuantParams{.scales={0.01f, 0.005f, 0.01f, 0.005f}, .channelDim=3}); in createTestModel_nchw_weight_as_input_channelQuant8() 3375 …ms1.channelQuant(SymmPerChannelQuantParams{.scales={0.01f, 0.005f, 0.01f, 0.005f}, .channelDim=3}); in createTestModel_nchw_weight_as_input_channelQuant8_2() 4139 …ms1.channelQuant(SymmPerChannelQuantParams{.scales={0.01f, 0.005f, 0.01f, 0.005f}, .channelDim=3}); in createTestModel_dynamic_output_shape_nhwc_channelQuant8() 4294 …ms1.channelQuant(SymmPerChannelQuantParams{.scales={0.01f, 0.005f, 0.01f, 0.005f}, .channelDim=3}); in createTestModel_dynamic_output_shape_nhwc_channelQuant8_2() [all …]
|
D | transpose_conv2d.model.cpp | 1008 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_none_channelQuant8() 1136 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_none_channelQuant8_weight_as_input() 1264 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_none_channelQuant8_2() 1392 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_none_channelQuant8_weight_as_input_2() 2772 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_relu_channelQuant8() 2900 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_relu_channelQuant8_weight_as_input() 3028 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_relu_channelQuant8_2() 3156 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_relu_channelQuant8_weight_as_input_2() 4536 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_relu1_channelQuant8() 4664 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.25f, 0.5f}, .channelDim=0}); in createTestModel_nhwc_relu1_channelQuant8_weight_as_input() [all …]
|
D | conv2d_v1_2.model.cpp | 436 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.125f}, .channelDim=0}); in createTestModel_nhwc_channelQuant8() 1155 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.125f}, .channelDim=0}); in createTestModel_nhwc_weight_as_input_channelQuant8() 1874 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.125f}, .channelDim=0}); in createTestModel_nchw_channelQuant8() 2593 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.125f}, .channelDim=0}); in createTestModel_nchw_weight_as_input_channelQuant8() 3312 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.125f}, .channelDim=0}); in createTestModel_dynamic_output_shape_nhwc_channelQuant8() 4031 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.125f}, .channelDim=0}); in createTestModel_dynamic_output_shape_nhwc_weight_as_input_channelQuant8() 4750 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.125f}, .channelDim=0}); in createTestModel_dynamic_output_shape_nchw_channelQuant8() 5469 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.125f}, .channelDim=0}); in createTestModel_dynamic_output_shape_nchw_weight_as_input_channelQuant8() 6107 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.5f}, .channelDim=0}); in createTestModel_nhwc_channelQuant8_2() 6691 extraParams1.channelQuant(SymmPerChannelQuantParams{.scales={0.5f}, .channelDim=0}); in createTestModel_nhwc_weight_as_input_channelQuant8_2() [all …]
|
/frameworks/ml/nn/common/ |
D | ValidateHal.cpp | 102 NN_RET_CHECK_LT(channelQuant.channelDim, count) in validateOperandExtraParams() 105 << " with an invalid channelQuant.channelDim " << channelQuant.channelDim in validateOperandExtraParams() 107 uint32_t expected = operand.dimensions[channelQuant.channelDim]; in validateOperandExtraParams() 115 << channelQuant.channelDim << " is underspecified (can't be 0)"; in validateOperandExtraParams()
|
D | Utils.cpp | 371 NN_RET_CHECK_LT(channelQuant.channelDim, halOperand.dimensions.size()) << tag; in validateOperandSymmPerChannelQuantParams() 373 NN_RET_CHECK_EQ(channelQuant.scaleCount, halOperand.dimensions[channelQuant.channelDim]) << tag; in validateOperandSymmPerChannelQuantParams() 374 NN_RET_CHECK_NE(halOperand.dimensions[channelQuant.channelDim], 0u) in validateOperandSymmPerChannelQuantParams() 375 << tag << " channel dimension " << channelQuant.channelDim << " is underspecified"; in validateOperandSymmPerChannelQuantParams() 376 for (uint32_t i = 0; i < halOperand.dimensions[channelQuant.channelDim]; i++) { in validateOperandSymmPerChannelQuantParams() 1615 operands[inputIndexes[1]].extraParams.channelQuant().channelDim != 0) { in validateOperation()
|
/frameworks/ml/nn/runtime/include/ |
D | NeuralNetworksWrapper.h | 73 SymmPerChannelQuantParams(std::vector<float> scalesVec, uint32_t channelDim) in SymmPerChannelQuantParams() 76 .channelDim = channelDim, in SymmPerChannelQuantParams()
|
D | NeuralNetworks.h | 5184 uint32_t channelDim; member
|
/frameworks/ml/nn/runtime/ |
D | ModelBuilder.cpp | 190 .channelDim = channelQuant.channelDim, in setOperandSymmPerChannelQuantParams()
|
D | ExecutionPlan.cpp | 185 .channelDim = fromChannelQuant.channelDim, in copyOperandExtraParams() 1385 return a.channelDim < b.channelDim; in initializeSlice()
|
/frameworks/ml/nn/tools/test_generator/ |
D | test_generator.py | 272 def __init__(self, channelDim, scales, hide = False): argument 273 self.channelDim = channelDim 279 bshape[self.channelDim] = len(self.scales) 284 ", ".join(str(x) + "f" for x in self.scales), self.channelDim) 291 ", ".join(str(x) + "f" for x in self.scales), self.channelDim)
|