Searched refs:op_params (Results 1 – 2 of 2) sorted by relevance
/packages/modules/NeuralNetworks/common/cpu_operations/ |
D | Broadcast.cpp | 169 tflite::ArithmeticParams op_params; in addQuant8() local 170 op_params.left_shift = left_shift; in addQuant8() 171 op_params.input1_offset = input1_offset; in addQuant8() 172 op_params.input1_multiplier = input1_multiplier; in addQuant8() 173 op_params.input1_shift = input1_shift; in addQuant8() 174 op_params.input2_offset = input2_offset; in addQuant8() 175 op_params.input2_multiplier = input2_multiplier; in addQuant8() 176 op_params.input2_shift = input2_shift; in addQuant8() 177 op_params.output_offset = output_offset; in addQuant8() 178 op_params.output_multiplier = output_multiplier; in addQuant8() [all …]
|
D | Pooling.cpp | 146 auto op_params = param.toTfliteParam(outputShape); in averagePoolNhwc() local 148 tflite::optimized_ops::AveragePool(op_params, convertShapeToTflshape(inputShape), inputData, in averagePoolNhwc() 169 auto op_params = param.toTfliteParam(outputShape); in averagePoolNhwc() local 171 tflite::optimized_ops::AveragePool(op_params, convertShapeToTflshape(inputShape), inputData, in averagePoolNhwc() 179 auto op_params = param.toTfliteParam(outputShape); in averagePoolNhwc() local 183 tflite::reference_integer_ops::AveragePool(op_params, convertShapeToTflshape(inputShape), in averagePoolNhwc() 192 auto op_params = param.toTfliteParam(outputShape); in l2PoolNhwc() local 194 tflite::optimized_ops::L2Pool(op_params, convertShapeToTflshape(inputShape), inputData, in l2PoolNhwc() 214 auto op_params = param.toTfliteParam(outputShape); in maxPoolNhwc() local 216 tflite::optimized_ops::MaxPool(op_params, convertShapeToTflshape(inputShape), inputData, in maxPoolNhwc() [all …]
|